flaxdiff 0.2.6__tar.gz → 0.2.6.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 (73) hide show
  1. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/PKG-INFO +1 -1
  2. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/inference/pipeline.py +2 -2
  3. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff.egg-info/PKG-INFO +1 -1
  4. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/pyproject.toml +1 -1
  5. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/README.md +0 -0
  6. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/__init__.py +0 -0
  7. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/data/__init__.py +0 -0
  8. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/data/benchmark_decord.py +0 -0
  9. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/data/dataloaders.py +0 -0
  10. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/data/dataset_map.py +0 -0
  11. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/data/online_loader.py +0 -0
  12. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/data/sources/audio_utils.py +0 -0
  13. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/data/sources/av_example.py +0 -0
  14. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/data/sources/av_utils.py +0 -0
  15. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/data/sources/base.py +0 -0
  16. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/data/sources/images.py +0 -0
  17. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/data/sources/utils.py +0 -0
  18. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/data/sources/videos.py +0 -0
  19. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/data/sources/voxceleb2.py +0 -0
  20. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/inference/__init__.py +0 -0
  21. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/inference/utils.py +0 -0
  22. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/inputs/__init__.py +0 -0
  23. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/inputs/encoders.py +0 -0
  24. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/metrics/__init__.py +0 -0
  25. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/metrics/common.py +0 -0
  26. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/metrics/images.py +0 -0
  27. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/metrics/inception.py +0 -0
  28. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/metrics/psnr.py +0 -0
  29. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/metrics/ssim.py +0 -0
  30. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/metrics/utils.py +0 -0
  31. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/models/__init__.py +0 -0
  32. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/models/attention.py +0 -0
  33. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/models/autoencoder/__init__.py +0 -0
  34. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/models/autoencoder/autoencoder.py +0 -0
  35. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/models/autoencoder/diffusers.py +0 -0
  36. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/models/autoencoder/simple_autoenc.py +0 -0
  37. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/models/better_uvit.py +0 -0
  38. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/models/common.py +0 -0
  39. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/models/favor_fastattn.py +0 -0
  40. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/models/general.py +0 -0
  41. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/models/simple_unet.py +0 -0
  42. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/models/simple_vit.py +0 -0
  43. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/models/unet_3d.py +0 -0
  44. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/models/unet_3d_blocks.py +0 -0
  45. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/predictors/__init__.py +0 -0
  46. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/samplers/__init__.py +0 -0
  47. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/samplers/common.py +0 -0
  48. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/samplers/ddim.py +0 -0
  49. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/samplers/ddpm.py +0 -0
  50. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/samplers/euler.py +0 -0
  51. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/samplers/heun_sampler.py +0 -0
  52. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/samplers/multistep_dpm.py +0 -0
  53. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/samplers/rk4_sampler.py +0 -0
  54. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/schedulers/__init__.py +0 -0
  55. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/schedulers/common.py +0 -0
  56. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/schedulers/continuous.py +0 -0
  57. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/schedulers/cosine.py +0 -0
  58. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/schedulers/discrete.py +0 -0
  59. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/schedulers/exp.py +0 -0
  60. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/schedulers/karras.py +0 -0
  61. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/schedulers/linear.py +0 -0
  62. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/schedulers/sqrt.py +0 -0
  63. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/trainer/__init__.py +0 -0
  64. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/trainer/autoencoder_trainer.py +0 -0
  65. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/trainer/diffusion_trainer.py +0 -0
  66. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/trainer/general_diffusion_trainer.py +0 -0
  67. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/trainer/simple_trainer.py +0 -0
  68. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/utils.py +0 -0
  69. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff.egg-info/SOURCES.txt +0 -0
  70. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff.egg-info/dependency_links.txt +0 -0
  71. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff.egg-info/requires.txt +0 -0
  72. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff.egg-info/top_level.txt +0 -0
  73. {flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/setup.cfg +0 -0
{flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: flaxdiff
3
- Version: 0.2.6
3
+ Version: 0.2.6.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.2.6 → flaxdiff-0.2.6.1}/flaxdiff/inference/pipeline.py RENAMED
@@ -20,7 +20,7 @@ from flaxdiff.models.autoencoder import AutoEncoder
20
20
  from flaxdiff.inputs import DiffusionInputConfig
21
21
  from flaxdiff.utils import defaultTextEncodeModel, RandomMarkovState
22
22
  from flaxdiff.samplers.euler import EulerAncestralSampler
23
- from .utils import parse_config, load_from_wandb_run, load_from_wandb_registry
23
+ from flaxdiff.inference.utils import parse_config, load_from_wandb_run, load_from_wandb_registry
24
24
 
25
25
  @dataclass
26
26
  class InferencePipeline:
@@ -51,9 +51,9 @@ class DiffusionInferencePipeline(InferencePipeline):
51
51
  model_output_transform: DiffusionPredictionTransform = None
52
52
  autoencoder: AutoEncoder = None
53
53
  input_config: DiffusionInputConfig = None
54
+ wandb_run = None
54
55
  samplers: Dict[Type[DiffusionSampler], Dict[float, DiffusionSampler]] = field(default_factory=dict)
55
56
  config: Dict[str, Any] = field(default_factory=dict)
56
- wandb_run = None
57
57
 
58
58
  @classmethod
59
59
  def from_wandb_run(
{flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: flaxdiff
3
- Version: 0.2.6
3
+ Version: 0.2.6.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.2.6 → flaxdiff-0.2.6.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.2.6"
7
+ version = "0.2.6.1"
8
8
  description = "A versatile and easy to understand Diffusion library"
9
9
  readme = "README.md"
10
10
  authors = [
{flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/README.md RENAMED
File without changes
{flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/flaxdiff/utils.py RENAMED
File without changes
{flaxdiff-0.2.6 → flaxdiff-0.2.6.1}/setup.cfg RENAMED
File without changes