flaxdiff 0.1.37__py3-none-any.whl → 0.1.37.2__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (6) hide show
  1. flaxdiff/samplers/__init__.py +1 -1
  2. flaxdiff/schedulers/__init__.py +6 -6
  3. {flaxdiff-0.1.37.dist-info → flaxdiff-0.1.37.2.dist-info}/METADATA +1 -1
  4. {flaxdiff-0.1.37.dist-info → flaxdiff-0.1.37.2.dist-info}/RECORD +6 -6
  5. {flaxdiff-0.1.37.dist-info → flaxdiff-0.1.37.2.dist-info}/WHEEL +0 -0
  6. {flaxdiff-0.1.37.dist-info → flaxdiff-0.1.37.2.dist-info}/top_level.txt +0 -0
flaxdiff/samplers/__init__.py CHANGED
@@ -1,7 +1,7 @@
1
1
  from .common import DiffusionSampler
2
2
  from .ddim import DDIMSampler
3
3
  from .ddpm import DDPMSampler, SimpleDDPMSampler
4
- from .euler import EulerSampler, SimplifiedEulerSampler
4
+ from .euler import EulerSampler, SimplifiedEulerSampler, EulerAncestralSampler
5
5
  from .heun_sampler import HeunSampler
6
6
  from .rk4_sampler import RK4Sampler
7
7
  from .multistep_dpm import MultiStepDPM
flaxdiff/schedulers/__init__.py CHANGED
@@ -1,6 +1,6 @@
1
- from .discrete import DiscreteNoiseScheduler
2
- from .common import NoiseScheduler, GeneralizedNoiseScheduler
3
- from .cosine import CosineNoiseScheduler, ContinuousNoiseScheduler, CosineGeneralNoiseScheduler
4
- from .linear import LinearNoiseSchedule
5
- from .sqrt import SqrtContinuousNoiseScheduler
6
- from .karras import KarrasVENoiseScheduler, SimpleExpNoiseScheduler, EDMNoiseScheduler
1
+ from .discrete import *
2
+ from .common import *
3
+ from .cosine import *
4
+ from .linear import *
5
+ from .sqrt import *
6
+ from .karras import *
{flaxdiff-0.1.37.dist-info → flaxdiff-0.1.37.2.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: flaxdiff
3
- Version: 0.1.37
3
+ Version: 0.1.37.2
4
4
  Summary: A versatile and easy to understand Diffusion library
5
5
  Author-email: Ashish Kumar Singh <ashishkmr472@gmail.com>
6
6
  License-Expression: MIT
{flaxdiff-0.1.37.dist-info → flaxdiff-0.1.37.2.dist-info}/RECORD RENAMED
@@ -19,7 +19,7 @@ flaxdiff/models/autoencoder/autoencoder.py,sha256=27_hYl0yXAdH9Mx4Xu9J79mSNo-FEK
19
19
  flaxdiff/models/autoencoder/diffusers.py,sha256=JHeFLCxiHhu-QHwhKiCuKsQJn4AZumquiuxgZkiYGQ0,3643
20
20
  flaxdiff/models/autoencoder/simple_autoenc.py,sha256=UXHPgDmwGTnv3Uts6Zj3p9R9nJXnEiEXbllgarwDfXM,805
21
21
  flaxdiff/predictors/__init__.py,sha256=SKkYYRF9Wfgk2zhtZw4vCXOdOeRlrm2Mk6cvuaEvAzc,4403
22
- flaxdiff/samplers/__init__.py,sha256=_S-9TwDeshrI0VmapV-J2hqjTByOa0-oOeUs_IdovjU,285
22
+ flaxdiff/samplers/__init__.py,sha256=EY9v1pgwEoR64Kiz9K8fAR-4_ir9c03mYeY3hrpUNhE,308
23
23
  flaxdiff/samplers/common.py,sha256=8SabB-lgcbJZtNBfS96cn2xakeOj6-HC0Evhzeh2NlY,8921
24
24
  flaxdiff/samplers/ddim.py,sha256=hTjDm0SmIj-Tkc80QRATMcN_sKVhHbqZQboRQCAn4mY,569
25
25
  flaxdiff/samplers/ddpm.py,sha256=JgkNSo7fp7Jm-8rCy4eu5m4YIzXTWzxv-iHf3EQ0z5w,2243
@@ -27,7 +27,7 @@ flaxdiff/samplers/euler.py,sha256=QSkttB4DYnepDGwhWq3EGXYjMAqj4qLOdh7u98HttzY,27
27
27
  flaxdiff/samplers/heun_sampler.py,sha256=EvR3hy4t_D47ZOH4luzRFqPmv2v4z78P_JhqBGEpHU8,1436
28
28
  flaxdiff/samplers/multistep_dpm.py,sha256=2M4Abb93-GUVN1f0_ZHBeA6lF0eF15Hi6QOgOu2K45s,2752
29
29
  flaxdiff/samplers/rk4_sampler.py,sha256=vcQefFhOUZdNOQGBdzNkb2NgmTC2KWd_nhUhyLtt3yI,2026
30
- flaxdiff/schedulers/__init__.py,sha256=EcABJ5UqsfeFXD9ypbgjVSYb6IKm7bcmsUbqEwVpHUc,376
30
+ flaxdiff/schedulers/__init__.py,sha256=EIva9gBz3DKHORuGmv1LQCKTtRqCRavFOXMNqxAR_ks,131
31
31
  flaxdiff/schedulers/common.py,sha256=b-W4iI-aqScpVE8VZbBpiYvAVI6rqDkUP-C_hEVBwCI,4151
32
32
  flaxdiff/schedulers/continuous.py,sha256=5c_niOA20fxJ5oJDi09FfayIRogBGwtfG0XThW2IUZk,334
33
33
  flaxdiff/schedulers/cosine.py,sha256=EtU3SjJaP9R9ULHNiYrX9jBLSsAGKPGteHiwOzWNzYo,2006
@@ -41,7 +41,7 @@ flaxdiff/trainer/autoencoder_trainer.py,sha256=hxihkRL9WCIQVGOP-pc1jjjIUaRXDLcNo
41
41
  flaxdiff/trainer/diffusion_trainer.py,sha256=KVeXJ9ZQKcvD-O_hCJnxro0dQRuQe5ZVGGMEL4Lgm9k,12814
42
42
  flaxdiff/trainer/simple_trainer.py,sha256=lmRo8N0bMupIyS3ejPvPtxoskY_3GLC8iyJE6u4TIWc,21990
43
43
  flaxdiff/trainer/video_diffusion_trainer.py,sha256=gMkKpnKNTo8QhTx5ptEEkc7W5-7rzXIr9queU53hXyQ,2197
44
- flaxdiff-0.1.37.dist-info/METADATA,sha256=T8xDHQRNMEXd0SvQZv4yYaSLjfpsYLdVytxHj9_VRZ4,23983
45
- flaxdiff-0.1.37.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
46
- flaxdiff-0.1.37.dist-info/top_level.txt,sha256=-2-nXnfkJgSfkki1tjm5Faw6Dso7vhtdn2szwCdX5CQ,9
47
- flaxdiff-0.1.37.dist-info/RECORD,,
44
+ flaxdiff-0.1.37.2.dist-info/METADATA,sha256=VxFaccvqQ8pZ72-_kXL_IlcP_18IqTGQZfiJtRdmNqc,23985
45
+ flaxdiff-0.1.37.2.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
46
+ flaxdiff-0.1.37.2.dist-info/top_level.txt,sha256=-2-nXnfkJgSfkki1tjm5Faw6Dso7vhtdn2szwCdX5CQ,9
47
+ flaxdiff-0.1.37.2.dist-info/RECORD,,