flaxdiff 0.1.37__tar.gz → 0.1.37.1__tar.gz

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 (54) hide show
  1. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/PKG-INFO +1 -1
  2. flaxdiff-0.1.37.1/flaxdiff/samplers/__init__.py +7 -0
  3. flaxdiff-0.1.37.1/flaxdiff/schedulers/__init__.py +6 -0
  4. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff.egg-info/PKG-INFO +1 -1
  5. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/pyproject.toml +1 -1
  6. flaxdiff-0.1.37/flaxdiff/samplers/__init__.py +0 -7
  7. flaxdiff-0.1.37/flaxdiff/schedulers/__init__.py +0 -6
  8. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/README.md +0 -0
  9. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/__init__.py +0 -0
  10. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/data/__init__.py +0 -0
  11. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/data/dataset_map.py +0 -0
  12. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/data/datasets.py +0 -0
  13. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/data/online_loader.py +0 -0
  14. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/data/sources/gcs.py +0 -0
  15. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/data/sources/tfds.py +0 -0
  16. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/metrics/inception.py +0 -0
  17. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/metrics/utils.py +0 -0
  18. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/models/__init__.py +0 -0
  19. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/models/attention.py +0 -0
  20. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/models/autoencoder/__init__.py +0 -0
  21. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/models/autoencoder/autoencoder.py +0 -0
  22. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/models/autoencoder/diffusers.py +0 -0
  23. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/models/autoencoder/simple_autoenc.py +0 -0
  24. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/models/common.py +0 -0
  25. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/models/favor_fastattn.py +0 -0
  26. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/models/simple_unet.py +0 -0
  27. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/models/simple_vit.py +0 -0
  28. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/predictors/__init__.py +0 -0
  29. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/samplers/common.py +0 -0
  30. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/samplers/ddim.py +0 -0
  31. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/samplers/ddpm.py +0 -0
  32. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/samplers/euler.py +0 -0
  33. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/samplers/heun_sampler.py +0 -0
  34. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/samplers/multistep_dpm.py +0 -0
  35. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/samplers/rk4_sampler.py +0 -0
  36. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/schedulers/common.py +0 -0
  37. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/schedulers/continuous.py +0 -0
  38. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/schedulers/cosine.py +0 -0
  39. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/schedulers/discrete.py +0 -0
  40. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/schedulers/exp.py +0 -0
  41. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/schedulers/karras.py +0 -0
  42. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/schedulers/linear.py +0 -0
  43. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/schedulers/sqrt.py +0 -0
  44. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/trainer/__init__.py +0 -0
  45. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/trainer/autoencoder_trainer.py +0 -0
  46. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/trainer/diffusion_trainer.py +0 -0
  47. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/trainer/simple_trainer.py +0 -0
  48. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/trainer/video_diffusion_trainer.py +0 -0
  49. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/utils.py +0 -0
  50. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff.egg-info/SOURCES.txt +0 -0
  51. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff.egg-info/dependency_links.txt +0 -0
  52. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff.egg-info/requires.txt +0 -0
  53. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff.egg-info/top_level.txt +0 -0
  54. {flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/setup.cfg +0 -0
{flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/PKG-INFO 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.1
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/flaxdiff/samplers/__init__.py ADDED
@@ -0,0 +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 *
flaxdiff-0.1.37.1/flaxdiff/schedulers/__init__.py ADDED
@@ -0,0 +1,6 @@
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 → flaxdiff-0.1.37.1}/flaxdiff.egg-info/PKG-INFO 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.1
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 → flaxdiff-0.1.37.1}/pyproject.toml RENAMED
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
4
4
 
5
5
  [project]
6
6
  name = "flaxdiff"
7
- version = "0.1.37"
7
+ version = "0.1.37.1"
8
8
  description = "A versatile and easy to understand Diffusion library"
9
9
  readme = "README.md"
10
10
  authors = [
flaxdiff-0.1.37/flaxdiff/samplers/__init__.py DELETED
@@ -1,7 +0,0 @@
1
- from .common import DiffusionSampler
2
- from .ddim import DDIMSampler
3
- from .ddpm import DDPMSampler, SimpleDDPMSampler
4
- from .euler import EulerSampler, SimplifiedEulerSampler
5
- from .heun_sampler import HeunSampler
6
- from .rk4_sampler import RK4Sampler
7
- from .multistep_dpm import MultiStepDPM
flaxdiff-0.1.37/flaxdiff/schedulers/__init__.py DELETED
@@ -1,6 +0,0 @@
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
{flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/README.md RENAMED
File without changes
{flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/flaxdiff/utils.py RENAMED
File without changes
{flaxdiff-0.1.37 → flaxdiff-0.1.37.1}/setup.cfg RENAMED
File without changes