flaxdiff 0.1.37.1__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 (5) hide show
  1. flaxdiff/samplers/__init__.py +7 -7
  2. {flaxdiff-0.1.37.1.dist-info → flaxdiff-0.1.37.2.dist-info}/METADATA +1 -1
  3. {flaxdiff-0.1.37.1.dist-info → flaxdiff-0.1.37.2.dist-info}/RECORD +5 -5
  4. {flaxdiff-0.1.37.1.dist-info → flaxdiff-0.1.37.2.dist-info}/WHEEL +0 -0
  5. {flaxdiff-0.1.37.1.dist-info → flaxdiff-0.1.37.2.dist-info}/top_level.txt +0 -0
flaxdiff/samplers/__init__.py CHANGED
@@ -1,7 +1,7 @@
1
- from .common import *
2
- from .ddim import *
3
- from .ddpm import *
4
- from .euler import *
5
- from .heun_sampler import *
6
- from .rk4_sampler import *
7
- from .multistep_dpm import *
1
+ from .common import DiffusionSampler
2
+ from .ddim import DDIMSampler
3
+ from .ddpm import DDPMSampler, SimpleDDPMSampler
4
+ from .euler import EulerSampler, SimplifiedEulerSampler, EulerAncestralSampler
5
+ from .heun_sampler import HeunSampler
6
+ from .rk4_sampler import RK4Sampler
7
+ from .multistep_dpm import MultiStepDPM
{flaxdiff-0.1.37.1.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.1
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.1.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=dsK_a6HwMpxNuDbl5SzmvKJHYC_DGbTxxS3gfQYwcUY,166
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
@@ -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.1.dist-info/METADATA,sha256=ThP8nRYIdfvEE2ytSgTbulqbCDBLaWk22t4u2Mk-7-8,23985
45
- flaxdiff-0.1.37.1.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
46
- flaxdiff-0.1.37.1.dist-info/top_level.txt,sha256=-2-nXnfkJgSfkki1tjm5Faw6Dso7vhtdn2szwCdX5CQ,9
47
- flaxdiff-0.1.37.1.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,,