vector-quantize-pytorch 1.22.0__tar.gz → 1.22.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 (32) hide show
  1. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/PKG-INFO +1 -1
  2. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/pyproject.toml +1 -1
  3. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/vector_quantize_pytorch/vector_quantize_pytorch.py +1 -0
  4. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/.github/workflows/build.yml +0 -0
  5. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/.github/workflows/python-publish.yml +0 -0
  6. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/.github/workflows/test.yml +0 -0
  7. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/.gitignore +0 -0
  8. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/LICENSE +0 -0
  9. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/README.md +0 -0
  10. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/examples/autoencoder.py +0 -0
  11. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/examples/autoencoder_fsq.py +0 -0
  12. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/examples/autoencoder_lfq.py +0 -0
  13. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/examples/autoencoder_sim_vq.py +0 -0
  14. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/images/fsq.png +0 -0
  15. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/images/lfq.png +0 -0
  16. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/images/simvq.png +0 -0
  17. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/images/vq.png +0 -0
  18. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/ruff.toml +0 -0
  19. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/tests/test_latent_quantization.py +0 -0
  20. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/tests/test_lfq.py +0 -0
  21. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/tests/test_readme.py +0 -0
  22. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/vector_quantize_pytorch/__init__.py +0 -0
  23. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/vector_quantize_pytorch/finite_scalar_quantization.py +0 -0
  24. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/vector_quantize_pytorch/latent_quantization.py +0 -0
  25. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/vector_quantize_pytorch/lookup_free_quantization.py +0 -0
  26. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/vector_quantize_pytorch/random_projection_quantizer.py +0 -0
  27. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/vector_quantize_pytorch/residual_fsq.py +0 -0
  28. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/vector_quantize_pytorch/residual_lfq.py +0 -0
  29. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/vector_quantize_pytorch/residual_sim_vq.py +0 -0
  30. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/vector_quantize_pytorch/residual_vq.py +0 -0
  31. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/vector_quantize_pytorch/sim_vq.py +0 -0
  32. {vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/vector_quantize_pytorch/utils.py +0 -0
{vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: vector-quantize-pytorch
3
- Version: 1.22.0
3
+ Version: 1.22.1
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
{vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "vector-quantize-pytorch"
3
- version = "1.22.0"
3
+ version = "1.22.1"
4
4
  description = "Vector Quantization - Pytorch"
5
5
  authors = [
6
6
  { name = "Phil Wang", email = "lucidrains@gmail.com" }
{vector_quantize_pytorch-1.22.0 → vector_quantize_pytorch-1.22.1}/vector_quantize_pytorch/vector_quantize_pytorch.py RENAMED
@@ -873,6 +873,7 @@ class VectorQuantize(Module):
873
873
  self.commitment_weight = commitment_weight
874
874
  self.commitment_use_cross_entropy_loss = commitment_use_cross_entropy_loss # whether to use cross entropy loss to codebook as commitment loss
875
875
 
876
+ assert not (use_cosine_sim and learnable_codebook), 'cosine sim distance codebook not compatible with learnable codebook yet'
876
877
  self.learnable_codebook = learnable_codebook
877
878
 
878
879
  has_codebook_orthogonal_loss = orthogonal_reg_weight > 0.