vector-quantize-pytorch 1.15.2__tar.gz → 1.15.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 (26) hide show
  1. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/PKG-INFO +10 -1
  2. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/README.md +9 -0
  3. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/pyproject.toml +1 -1
  4. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/vector_quantize_pytorch/vector_quantize_pytorch.py +2 -2
  5. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/.github/workflows/build.yml +0 -0
  6. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/.github/workflows/python-publish.yml +0 -0
  7. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/.github/workflows/test.yml +0 -0
  8. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/.gitignore +0 -0
  9. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/LICENSE +0 -0
  10. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/examples/autoencoder.py +0 -0
  11. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/examples/autoencoder_fsq.py +0 -0
  12. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/examples/autoencoder_lfq.py +0 -0
  13. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/images/fsq.png +0 -0
  14. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/images/lfq.png +0 -0
  15. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/images/vq.png +0 -0
  16. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/ruff.toml +0 -0
  17. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/tests/test_latent_quantization.py +0 -0
  18. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/tests/test_readme.py +0 -0
  19. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/vector_quantize_pytorch/__init__.py +0 -0
  20. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/vector_quantize_pytorch/finite_scalar_quantization.py +0 -0
  21. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/vector_quantize_pytorch/latent_quantization.py +0 -0
  22. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/vector_quantize_pytorch/lookup_free_quantization.py +0 -0
  23. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/vector_quantize_pytorch/random_projection_quantizer.py +0 -0
  24. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/vector_quantize_pytorch/residual_fsq.py +0 -0
  25. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/vector_quantize_pytorch/residual_lfq.py +0 -0
  26. {vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/vector_quantize_pytorch/residual_vq.py +0 -0
{vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: vector-quantize-pytorch
3
- Version: 1.15.2
3
+ Version: 1.15.3
4
4
  Summary: Vector Quantization - Pytorch
5
5
  Project-URL: Homepage, https://pypi.org/project/vector-quantize-pytorch/
6
6
  Project-URL: Repository, https://github.com/lucidrains/vector-quantizer-pytorch
@@ -723,3 +723,12 @@ assert loss.item() >= 0
723
723
  primaryClass = {cs.LG}
724
724
  }
725
725
  ```
726
+
727
+ ```bibtex
728
+ @inproceedings{Irie2023SelfOrganisingND,
729
+ title = {Self-Organising Neural Discrete Representation Learning \`a la Kohonen},
730
+ author = {Kazuki Irie and R'obert Csord'as and J{\"u}rgen Schmidhuber},
731
+ year = {2023},
732
+ url = {https://api.semanticscholar.org/CorpusID:256901024}
733
+ }
734
+ ```
{vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/README.md RENAMED
@@ -679,3 +679,12 @@ assert loss.item() >= 0
679
679
  primaryClass = {cs.LG}
680
680
  }
681
681
  ```
682
+
683
+ ```bibtex
684
+ @inproceedings{Irie2023SelfOrganisingND,
685
+ title = {Self-Organising Neural Discrete Representation Learning \`a la Kohonen},
686
+ author = {Kazuki Irie and R'obert Csord'as and J{\"u}rgen Schmidhuber},
687
+ year = {2023},
688
+ url = {https://api.semanticscholar.org/CorpusID:256901024}
689
+ }
690
+ ```
{vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "vector-quantize-pytorch"
3
- version = "1.15.2"
3
+ version = "1.15.3"
4
4
  description = "Vector Quantization - Pytorch"
5
5
  authors = [
6
6
  { name = "Phil Wang", email = "lucidrains@gmail.com" }
{vector_quantize_pytorch-1.15.2 → vector_quantize_pytorch-1.15.3}/vector_quantize_pytorch/vector_quantize_pytorch.py RENAMED
@@ -312,7 +312,7 @@ class EuclideanCodebook(Module):
312
312
  self.all_reduce_fn = distributed.all_reduce if use_ddp else noop
313
313
 
314
314
  self.register_buffer('initted', torch.Tensor([not kmeans_init]))
315
- self.register_buffer('cluster_size', torch.zeros(num_codebooks, codebook_size))
315
+ self.register_buffer('cluster_size', torch.ones(num_codebooks, codebook_size))
316
316
  self.register_buffer('embed_avg', embed.clone())
317
317
 
318
318
  self.learnable_codebook = learnable_codebook
@@ -582,7 +582,7 @@ class CosineSimCodebook(Module):
582
582
  self.all_reduce_fn = distributed.all_reduce if use_ddp else noop
583
583
 
584
584
  self.register_buffer('initted', torch.Tensor([not kmeans_init]))
585
- self.register_buffer('cluster_size', torch.zeros(num_codebooks, codebook_size))
585
+ self.register_buffer('cluster_size', torch.ones(num_codebooks, codebook_size))
586
586
  self.register_buffer('embed_avg', embed.clone())
587
587
 
588
588
  self.learnable_codebook = learnable_codebook