flaxdiff 0.2.1__tar.gz → 0.2.2__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 (69) hide show
  1. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/PKG-INFO +1 -1
  2. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/trainer/general_diffusion_trainer.py +4 -3
  3. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff.egg-info/PKG-INFO +1 -1
  4. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/pyproject.toml +1 -1
  5. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/README.md +0 -0
  6. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/__init__.py +0 -0
  7. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/data/__init__.py +0 -0
  8. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/data/benchmark_decord.py +0 -0
  9. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/data/dataloaders.py +0 -0
  10. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/data/dataset_map.py +0 -0
  11. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/data/online_loader.py +0 -0
  12. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/data/sources/audio_utils.py +0 -0
  13. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/data/sources/av_example.py +0 -0
  14. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/data/sources/av_utils.py +0 -0
  15. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/data/sources/base.py +0 -0
  16. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/data/sources/images.py +0 -0
  17. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/data/sources/utils.py +0 -0
  18. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/data/sources/videos.py +0 -0
  19. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/data/sources/voxceleb2.py +0 -0
  20. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/inference/__init__.py +0 -0
  21. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/inference/pipeline.py +0 -0
  22. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/inference/utils.py +0 -0
  23. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/inputs/__init__.py +0 -0
  24. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/inputs/encoders.py +0 -0
  25. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/metrics/inception.py +0 -0
  26. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/metrics/psnr.py +0 -0
  27. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/metrics/ssim.py +0 -0
  28. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/metrics/utils.py +0 -0
  29. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/models/__init__.py +0 -0
  30. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/models/attention.py +0 -0
  31. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/models/autoencoder/__init__.py +0 -0
  32. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/models/autoencoder/autoencoder.py +0 -0
  33. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/models/autoencoder/diffusers.py +0 -0
  34. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/models/autoencoder/simple_autoenc.py +0 -0
  35. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/models/common.py +0 -0
  36. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/models/favor_fastattn.py +0 -0
  37. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/models/general.py +0 -0
  38. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/models/simple_unet.py +0 -0
  39. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/models/simple_vit.py +0 -0
  40. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/models/unet_3d.py +0 -0
  41. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/models/unet_3d_blocks.py +0 -0
  42. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/predictors/__init__.py +0 -0
  43. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/samplers/__init__.py +0 -0
  44. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/samplers/common.py +0 -0
  45. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/samplers/ddim.py +0 -0
  46. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/samplers/ddpm.py +0 -0
  47. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/samplers/euler.py +0 -0
  48. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/samplers/heun_sampler.py +0 -0
  49. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/samplers/multistep_dpm.py +0 -0
  50. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/samplers/rk4_sampler.py +0 -0
  51. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/schedulers/__init__.py +0 -0
  52. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/schedulers/common.py +0 -0
  53. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/schedulers/continuous.py +0 -0
  54. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/schedulers/cosine.py +0 -0
  55. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/schedulers/discrete.py +0 -0
  56. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/schedulers/exp.py +0 -0
  57. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/schedulers/karras.py +0 -0
  58. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/schedulers/linear.py +0 -0
  59. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/schedulers/sqrt.py +0 -0
  60. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/trainer/__init__.py +0 -0
  61. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/trainer/autoencoder_trainer.py +0 -0
  62. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/trainer/diffusion_trainer.py +0 -0
  63. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/trainer/simple_trainer.py +0 -0
  64. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/utils.py +0 -0
  65. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff.egg-info/SOURCES.txt +0 -0
  66. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff.egg-info/dependency_links.txt +0 -0
  67. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff.egg-info/requires.txt +0 -0
  68. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff.egg-info/top_level.txt +0 -0
  69. {flaxdiff-0.2.1 → flaxdiff-0.2.2}/setup.cfg +0 -0
{flaxdiff-0.2.1 → flaxdiff-0.2.2}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: flaxdiff
3
- Version: 0.2.1
3
+ Version: 0.2.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.2.1 → flaxdiff-0.2.2}/flaxdiff/trainer/general_diffusion_trainer.py RENAMED
@@ -484,7 +484,7 @@ class GeneralDiffusionTrainer(DiffusionTrainer):
484
484
  def push_to_registry(
485
485
  self,
486
486
  registry_name: str = 'wandb-registry-model',
487
- aliases: List[str] = ['latest'],
487
+ aliases: List[str] = [],
488
488
  ):
489
489
  """
490
490
  Push the model to wandb registry.
@@ -504,7 +504,7 @@ class GeneralDiffusionTrainer(DiffusionTrainer):
504
504
  artifact_or_path=latest_checkpoint_path,
505
505
  name=modelname,
506
506
  type="model",
507
- aliases=aliases,
507
+ aliases=['latest'] + aliases,
508
508
  )
509
509
 
510
510
  target_path = f"{registry_name}/{modelname}"
@@ -512,6 +512,7 @@ class GeneralDiffusionTrainer(DiffusionTrainer):
512
512
  self.wandb.link_artifact(
513
513
  artifact=logged_artifact,
514
514
  target_path=target_path,
515
+ aliases=aliases,
515
516
  )
516
517
  print(f"Model pushed to registry at {target_path}")
517
518
  return logged_artifact
@@ -582,7 +583,7 @@ class GeneralDiffusionTrainer(DiffusionTrainer):
582
583
  is_good, is_best = self.__compare_run_against_best__(top_k=5, metric="train/best_loss")
583
584
  if is_good:
584
585
  # Push to registry with appropriate aliases
585
- aliases = ["latest"]
586
+ aliases = []
586
587
  if is_best:
587
588
  aliases.append("best")
588
589
  self.push_to_registry(aliases=aliases)
{flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: flaxdiff
3
- Version: 0.2.1
3
+ Version: 0.2.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.2.1 → flaxdiff-0.2.2}/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.1"
7
+ version = "0.2.2"
8
8
  description = "A versatile and easy to understand Diffusion library"
9
9
  readme = "README.md"
10
10
  authors = [
{flaxdiff-0.2.1 → flaxdiff-0.2.2}/README.md RENAMED
File without changes
{flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/__init__.py RENAMED
File without changes
{flaxdiff-0.2.1 → flaxdiff-0.2.2}/flaxdiff/utils.py RENAMED
File without changes
{flaxdiff-0.2.1 → flaxdiff-0.2.2}/setup.cfg RENAMED
File without changes