flaxdiff 0.1.35.3__tar.gz → 0.1.35.4__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 (45) hide show
  1. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/PKG-INFO +1 -1
  2. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/data/online_loader.py +2 -2
  3. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/trainer/simple_trainer.py +1 -0
  4. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff.egg-info/PKG-INFO +1 -1
  5. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/setup.py +1 -1
  6. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/README.md +0 -0
  7. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/__init__.py +0 -0
  8. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/data/__init__.py +0 -0
  9. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/models/__init__.py +0 -0
  10. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/models/attention.py +0 -0
  11. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/models/autoencoder/__init__.py +0 -0
  12. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/models/autoencoder/autoencoder.py +0 -0
  13. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/models/autoencoder/diffusers.py +0 -0
  14. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/models/autoencoder/simple_autoenc.py +0 -0
  15. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/models/common.py +0 -0
  16. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/models/favor_fastattn.py +0 -0
  17. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/models/simple_unet.py +0 -0
  18. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/models/simple_vit.py +0 -0
  19. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/predictors/__init__.py +0 -0
  20. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/samplers/__init__.py +0 -0
  21. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/samplers/common.py +0 -0
  22. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/samplers/ddim.py +0 -0
  23. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/samplers/ddpm.py +0 -0
  24. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/samplers/euler.py +0 -0
  25. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/samplers/heun_sampler.py +0 -0
  26. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/samplers/multistep_dpm.py +0 -0
  27. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/samplers/rk4_sampler.py +0 -0
  28. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/schedulers/__init__.py +0 -0
  29. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/schedulers/common.py +0 -0
  30. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/schedulers/continuous.py +0 -0
  31. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/schedulers/cosine.py +0 -0
  32. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/schedulers/discrete.py +0 -0
  33. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/schedulers/exp.py +0 -0
  34. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/schedulers/karras.py +0 -0
  35. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/schedulers/linear.py +0 -0
  36. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/schedulers/sqrt.py +0 -0
  37. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/trainer/__init__.py +0 -0
  38. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/trainer/autoencoder_trainer.py +0 -0
  39. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/trainer/diffusion_trainer.py +0 -0
  40. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/utils.py +0 -0
  41. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff.egg-info/SOURCES.txt +0 -0
  42. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff.egg-info/dependency_links.txt +0 -0
  43. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff.egg-info/requires.txt +0 -0
  44. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff.egg-info/top_level.txt +0 -0
  45. {flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/setup.cfg +0 -0
{flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: flaxdiff
3
- Version: 0.1.35.3
3
+ Version: 0.1.35.4
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.35.3 → flaxdiff-0.1.35.4}/flaxdiff/data/online_loader.py RENAMED
@@ -111,8 +111,8 @@ def map_sample(
111
111
  "original_width": original_width,
112
112
  })
113
113
  except Exception as e:
114
- print(f"Error maping sample {url}", e)
115
- traceback.print_exc()
114
+ # print(f"Error maping sample {url}", e)
115
+ # traceback.print_exc()
116
116
  # error_queue.put_nowait({
117
117
  # "url": url,
118
118
  # "caption": caption,
{flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff/trainer/simple_trainer.py RENAMED
@@ -437,5 +437,6 @@ class SimpleTrainer:
437
437
  "train/epoch": current_epoch,
438
438
  }, step=current_step)
439
439
  print(colored(f"\n\tEpoch {current_epoch} completed. Avg Loss: {avg_loss}, Time: {total_time:.2f}s, Best Loss: {self.best_loss}", 'green'))
440
+ print("Training done")
440
441
  self.save(epochs)
441
442
  return self.state
{flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/flaxdiff.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: flaxdiff
3
- Version: 0.1.35.3
3
+ Version: 0.1.35.4
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.35.3 → flaxdiff-0.1.35.4}/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.35.3',
14
+ version='0.1.35.4',
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.35.3 → flaxdiff-0.1.35.4}/README.md RENAMED
File without changes
{flaxdiff-0.1.35.3 → flaxdiff-0.1.35.4}/setup.cfg RENAMED
File without changes