flaxdiff 0.1.36.2__tar.gz → 0.1.36.3__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.36.2 → flaxdiff-0.1.36.3}/PKG-INFO +1 -1
  2. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3}/flaxdiff.egg-info/PKG-INFO +1 -1
  3. flaxdiff-0.1.36.3/flaxdiff.egg-info/SOURCES.txt +50 -0
  4. flaxdiff-0.1.36.3/flaxdiff.egg-info/top_level.txt +1 -0
  5. {flaxdiff-0.1.36.2 → flaxdiff-0.1.36.3}/pyproject.toml +5 -2
  6. flaxdiff-0.1.36.2/src/flaxdiff.egg-info/SOURCES.txt +0 -50
  7. flaxdiff-0.1.36.2/src/flaxdiff.egg-info/top_level.txt +0 -9
  8. {flaxdiff-0.1.36.2 → flaxdiff-0.1.36.3}/README.md +0 -0
  9. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/__init__.py +0 -0
  10. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/data/__init__.py +0 -0
  11. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/data/dataset_map.py +0 -0
  12. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/data/datasets.py +0 -0
  13. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/data/online_loader.py +0 -0
  14. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/data/sources/gcs.py +0 -0
  15. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/data/sources/tfds.py +0 -0
  16. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/metrics/inception.py +0 -0
  17. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/metrics/utils.py +0 -0
  18. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/models/__init__.py +0 -0
  19. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/models/attention.py +0 -0
  20. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/models/autoencoder/__init__.py +0 -0
  21. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/models/autoencoder/autoencoder.py +0 -0
  22. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/models/autoencoder/diffusers.py +0 -0
  23. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/models/autoencoder/simple_autoenc.py +0 -0
  24. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/models/common.py +0 -0
  25. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/models/favor_fastattn.py +0 -0
  26. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/models/simple_unet.py +0 -0
  27. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/models/simple_vit.py +0 -0
  28. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/predictors/__init__.py +0 -0
  29. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/samplers/__init__.py +0 -0
  30. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/samplers/common.py +0 -0
  31. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/samplers/ddim.py +0 -0
  32. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/samplers/ddpm.py +0 -0
  33. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/samplers/euler.py +0 -0
  34. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/samplers/heun_sampler.py +0 -0
  35. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/samplers/multistep_dpm.py +0 -0
  36. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/samplers/rk4_sampler.py +0 -0
  37. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/schedulers/__init__.py +0 -0
  38. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/schedulers/common.py +0 -0
  39. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/schedulers/continuous.py +0 -0
  40. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/schedulers/cosine.py +0 -0
  41. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/schedulers/discrete.py +0 -0
  42. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/schedulers/exp.py +0 -0
  43. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/schedulers/karras.py +0 -0
  44. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/schedulers/linear.py +0 -0
  45. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/schedulers/sqrt.py +0 -0
  46. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/trainer/__init__.py +0 -0
  47. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/trainer/autoencoder_trainer.py +0 -0
  48. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/trainer/diffusion_trainer.py +0 -0
  49. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/trainer/simple_trainer.py +0 -0
  50. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/trainer/video_diffusion_trainer.py +0 -0
  51. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3/flaxdiff}/utils.py +0 -0
  52. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3}/flaxdiff.egg-info/dependency_links.txt +0 -0
  53. {flaxdiff-0.1.36.2/src → flaxdiff-0.1.36.3}/flaxdiff.egg-info/requires.txt +0 -0
  54. {flaxdiff-0.1.36.2 → flaxdiff-0.1.36.3}/setup.cfg +0 -0
{flaxdiff-0.1.36.2 → flaxdiff-0.1.36.3}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: flaxdiff
3
- Version: 0.1.36.2
3
+ Version: 0.1.36.3
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.36.2/src → flaxdiff-0.1.36.3}/flaxdiff.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: flaxdiff
3
- Version: 0.1.36.2
3
+ Version: 0.1.36.3
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.36.3/flaxdiff.egg-info/SOURCES.txt ADDED
@@ -0,0 +1,50 @@
1
+ README.md
2
+ pyproject.toml
3
+ flaxdiff/__init__.py
4
+ flaxdiff/utils.py
5
+ flaxdiff.egg-info/PKG-INFO
6
+ flaxdiff.egg-info/SOURCES.txt
7
+ flaxdiff.egg-info/dependency_links.txt
8
+ flaxdiff.egg-info/requires.txt
9
+ flaxdiff.egg-info/top_level.txt
10
+ flaxdiff/data/__init__.py
11
+ flaxdiff/data/dataset_map.py
12
+ flaxdiff/data/datasets.py
13
+ flaxdiff/data/online_loader.py
14
+ flaxdiff/data/sources/gcs.py
15
+ flaxdiff/data/sources/tfds.py
16
+ flaxdiff/metrics/inception.py
17
+ flaxdiff/metrics/utils.py
18
+ flaxdiff/models/__init__.py
19
+ flaxdiff/models/attention.py
20
+ flaxdiff/models/common.py
21
+ flaxdiff/models/favor_fastattn.py
22
+ flaxdiff/models/simple_unet.py
23
+ flaxdiff/models/simple_vit.py
24
+ flaxdiff/models/autoencoder/__init__.py
25
+ flaxdiff/models/autoencoder/autoencoder.py
26
+ flaxdiff/models/autoencoder/diffusers.py
27
+ flaxdiff/models/autoencoder/simple_autoenc.py
28
+ flaxdiff/predictors/__init__.py
29
+ flaxdiff/samplers/__init__.py
30
+ flaxdiff/samplers/common.py
31
+ flaxdiff/samplers/ddim.py
32
+ flaxdiff/samplers/ddpm.py
33
+ flaxdiff/samplers/euler.py
34
+ flaxdiff/samplers/heun_sampler.py
35
+ flaxdiff/samplers/multistep_dpm.py
36
+ flaxdiff/samplers/rk4_sampler.py
37
+ flaxdiff/schedulers/__init__.py
38
+ flaxdiff/schedulers/common.py
39
+ flaxdiff/schedulers/continuous.py
40
+ flaxdiff/schedulers/cosine.py
41
+ flaxdiff/schedulers/discrete.py
42
+ flaxdiff/schedulers/exp.py
43
+ flaxdiff/schedulers/karras.py
44
+ flaxdiff/schedulers/linear.py
45
+ flaxdiff/schedulers/sqrt.py
46
+ flaxdiff/trainer/__init__.py
47
+ flaxdiff/trainer/autoencoder_trainer.py
48
+ flaxdiff/trainer/diffusion_trainer.py
49
+ flaxdiff/trainer/simple_trainer.py
50
+ flaxdiff/trainer/video_diffusion_trainer.py
flaxdiff-0.1.36.3/flaxdiff.egg-info/top_level.txt ADDED
@@ -0,0 +1 @@
1
+ flaxdiff
{flaxdiff-0.1.36.2 → flaxdiff-0.1.36.3}/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.36.2"
7
+ version = "0.1.36.3"
8
8
  description = "A versatile and easy to understand Diffusion library"
9
9
  readme = "README.md"
10
10
  authors = [
@@ -26,4 +26,7 @@ dependencies = [
26
26
  "rich",
27
27
  "python-dotenv",
28
28
  ]
29
- license = "MIT"
29
+ license = "MIT"
30
+
31
+ [tool.setuptools.packages.find]
32
+ include = ["flaxdiff*"]
flaxdiff-0.1.36.2/src/flaxdiff.egg-info/SOURCES.txt DELETED
@@ -1,50 +0,0 @@
1
- README.md
2
- pyproject.toml
3
- src/__init__.py
4
- src/utils.py
5
- src/data/__init__.py
6
- src/data/dataset_map.py
7
- src/data/datasets.py
8
- src/data/online_loader.py
9
- src/data/sources/gcs.py
10
- src/data/sources/tfds.py
11
- src/flaxdiff.egg-info/PKG-INFO
12
- src/flaxdiff.egg-info/SOURCES.txt
13
- src/flaxdiff.egg-info/dependency_links.txt
14
- src/flaxdiff.egg-info/requires.txt
15
- src/flaxdiff.egg-info/top_level.txt
16
- src/metrics/inception.py
17
- src/metrics/utils.py
18
- src/models/__init__.py
19
- src/models/attention.py
20
- src/models/common.py
21
- src/models/favor_fastattn.py
22
- src/models/simple_unet.py
23
- src/models/simple_vit.py
24
- src/models/autoencoder/__init__.py
25
- src/models/autoencoder/autoencoder.py
26
- src/models/autoencoder/diffusers.py
27
- src/models/autoencoder/simple_autoenc.py
28
- src/predictors/__init__.py
29
- src/samplers/__init__.py
30
- src/samplers/common.py
31
- src/samplers/ddim.py
32
- src/samplers/ddpm.py
33
- src/samplers/euler.py
34
- src/samplers/heun_sampler.py
35
- src/samplers/multistep_dpm.py
36
- src/samplers/rk4_sampler.py
37
- src/schedulers/__init__.py
38
- src/schedulers/common.py
39
- src/schedulers/continuous.py
40
- src/schedulers/cosine.py
41
- src/schedulers/discrete.py
42
- src/schedulers/exp.py
43
- src/schedulers/karras.py
44
- src/schedulers/linear.py
45
- src/schedulers/sqrt.py
46
- src/trainer/__init__.py
47
- src/trainer/autoencoder_trainer.py
48
- src/trainer/diffusion_trainer.py
49
- src/trainer/simple_trainer.py
50
- src/trainer/video_diffusion_trainer.py
flaxdiff-0.1.36.2/src/flaxdiff.egg-info/top_level.txt DELETED
@@ -1,9 +0,0 @@
1
- __init__
2
- data
3
- metrics
4
- models
5
- predictors
6
- samplers
7
- schedulers
8
- trainer
9
- utils
{flaxdiff-0.1.36.2 → flaxdiff-0.1.36.3}/README.md RENAMED
File without changes
{flaxdiff-0.1.36.2 → flaxdiff-0.1.36.3}/setup.cfg RENAMED
File without changes