flaxdiff 0.1.6__tar.gz → 0.1.7__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 (43) hide show
  1. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/PKG-INFO +1 -1
  2. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/models/simple_unet.py +1 -1
  3. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff.egg-info/PKG-INFO +1 -1
  4. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/setup.py +1 -1
  5. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/README.md +0 -0
  6. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/__init__.py +0 -0
  7. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/models/__init__.py +0 -0
  8. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/models/attention.py +0 -0
  9. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/models/autoencoder/__init__.py +0 -0
  10. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/models/autoencoder/autoencoder.py +0 -0
  11. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/models/autoencoder/diffusers.py +0 -0
  12. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/models/autoencoder/simple_autoenc.py +0 -0
  13. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/models/common.py +0 -0
  14. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/models/favor_fastattn.py +0 -0
  15. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/models/simple_vit.py +0 -0
  16. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/predictors/__init__.py +0 -0
  17. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/samplers/__init__.py +0 -0
  18. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/samplers/common.py +0 -0
  19. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/samplers/ddim.py +0 -0
  20. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/samplers/ddpm.py +0 -0
  21. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/samplers/euler.py +0 -0
  22. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/samplers/heun_sampler.py +0 -0
  23. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/samplers/multistep_dpm.py +0 -0
  24. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/samplers/rk4_sampler.py +0 -0
  25. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/schedulers/__init__.py +0 -0
  26. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/schedulers/common.py +0 -0
  27. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/schedulers/continuous.py +0 -0
  28. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/schedulers/cosine.py +0 -0
  29. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/schedulers/discrete.py +0 -0
  30. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/schedulers/exp.py +0 -0
  31. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/schedulers/karras.py +0 -0
  32. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/schedulers/linear.py +0 -0
  33. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/schedulers/sqrt.py +0 -0
  34. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/trainer/__init__.py +0 -0
  35. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/trainer/autoencoder_trainer.py +0 -0
  36. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/trainer/diffusion_trainer.py +0 -0
  37. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/trainer/simple_trainer.py +0 -0
  38. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/utils.py +0 -0
  39. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff.egg-info/SOURCES.txt +0 -0
  40. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff.egg-info/dependency_links.txt +0 -0
  41. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff.egg-info/requires.txt +0 -0
  42. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff.egg-info/top_level.txt +0 -0
  43. {flaxdiff-0.1.6 → flaxdiff-0.1.7}/setup.cfg +0 -0
{flaxdiff-0.1.6 → flaxdiff-0.1.7}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: flaxdiff
3
- Version: 0.1.6
3
+ Version: 0.1.7
4
4
  Summary: A versatile and easy to understand Diffusion library
5
5
  Author: Ashish Kumar Singh
6
6
  Author-email: ashishkmr472@gmail.com
{flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/models/simple_unet.py RENAMED
@@ -4,7 +4,7 @@ from flax import linen as nn
4
4
  from flax.typing import Dtype, PrecisionLike
5
5
  from typing import Dict, Callable, Sequence, Any, Union, Optional
6
6
  import einops
7
- from .common import kernel_init, ConvLayer, Downsample, Upsample, FourierEmbedding, TimeProjection
7
+ from .common import kernel_init, ConvLayer, Downsample, Upsample, FourierEmbedding, TimeProjection, ResidualBlock
8
8
  from .attention import TransformerBlock
9
9
 
10
10
  class Unet(nn.Module):
{flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: flaxdiff
3
- Version: 0.1.6
3
+ Version: 0.1.7
4
4
  Summary: A versatile and easy to understand Diffusion library
5
5
  Author: Ashish Kumar Singh
6
6
  Author-email: ashishkmr472@gmail.com
{flaxdiff-0.1.6 → flaxdiff-0.1.7}/setup.py RENAMED
@@ -11,7 +11,7 @@ required_packages=[
11
11
  setup(
12
12
  name='flaxdiff',
13
13
  packages=find_packages(),
14
- version='0.1.6',
14
+ version='0.1.7',
15
15
  description='A versatile and easy to understand Diffusion library',
16
16
  long_description=open('README.md').read(),
17
17
  long_description_content_type='text/markdown',
{flaxdiff-0.1.6 → flaxdiff-0.1.7}/README.md RENAMED
File without changes
{flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/__init__.py RENAMED
File without changes
{flaxdiff-0.1.6 → flaxdiff-0.1.7}/flaxdiff/utils.py RENAMED
File without changes
{flaxdiff-0.1.6 → flaxdiff-0.1.7}/setup.cfg RENAMED
File without changes