vector-quantize-pytorch 1.19.2__tar.gz → 1.19.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.19.2 → vector_quantize_pytorch-1.19.3}/PKG-INFO +1 -1
  2. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/pyproject.toml +1 -1
  3. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/vector_quantize_pytorch/residual_fsq.py +1 -1
  4. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/vector_quantize_pytorch/residual_lfq.py +1 -1
  5. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/vector_quantize_pytorch/residual_vq.py +1 -2
  6. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/.github/workflows/build.yml +0 -0
  7. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/.github/workflows/python-publish.yml +0 -0
  8. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/.github/workflows/test.yml +0 -0
  9. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/.gitignore +0 -0
  10. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/LICENSE +0 -0
  11. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/README.md +0 -0
  12. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/examples/autoencoder.py +0 -0
  13. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/examples/autoencoder_fsq.py +0 -0
  14. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/examples/autoencoder_lfq.py +0 -0
  15. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/images/fsq.png +0 -0
  16. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/images/lfq.png +0 -0
  17. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/images/vq.png +0 -0
  18. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/ruff.toml +0 -0
  19. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/tests/test_latent_quantization.py +0 -0
  20. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/tests/test_readme.py +0 -0
  21. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/vector_quantize_pytorch/__init__.py +0 -0
  22. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/vector_quantize_pytorch/finite_scalar_quantization.py +0 -0
  23. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/vector_quantize_pytorch/latent_quantization.py +0 -0
  24. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/vector_quantize_pytorch/lookup_free_quantization.py +0 -0
  25. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/vector_quantize_pytorch/random_projection_quantizer.py +0 -0
  26. {vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/vector_quantize_pytorch/vector_quantize_pytorch.py +0 -0
{vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: vector-quantize-pytorch
3
- Version: 1.19.2
3
+ Version: 1.19.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
{vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "vector-quantize-pytorch"
3
- version = "1.19.2"
3
+ version = "1.19.3"
4
4
  description = "Vector Quantization - Pytorch"
5
5
  authors = [
6
6
  { name = "Phil Wang", email = "lucidrains@gmail.com" }
{vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/vector_quantize_pytorch/residual_fsq.py RENAMED
@@ -305,7 +305,7 @@ class GroupedResidualFSQ(Module):
305
305
 
306
306
  forward_kwargs = dict(
307
307
  return_all_codes = return_all_codes,
308
- rand_quantize_dropout_fixed_seed = get_maybe_sync_seed(device)
308
+ rand_quantize_dropout_fixed_seed = get_maybe_sync_seed(device) if self.training else None
309
309
  )
310
310
 
311
311
  # invoke residual vq on each group
{vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/vector_quantize_pytorch/residual_lfq.py RENAMED
@@ -272,7 +272,7 @@ class GroupedResidualLFQ(Module):
272
272
  forward_kwargs = dict(
273
273
  mask = mask,
274
274
  return_all_codes = return_all_codes,
275
- rand_quantize_dropout_fixed_seed = get_maybe_sync_seed(device)
275
+ rand_quantize_dropout_fixed_seed = get_maybe_sync_seed(device) if self.training else None
276
276
  )
277
277
 
278
278
  # invoke residual vq on each group
{vector_quantize_pytorch-1.19.2 → vector_quantize_pytorch-1.19.3}/vector_quantize_pytorch/residual_vq.py RENAMED
@@ -328,7 +328,6 @@ class ResidualVQ(Module):
328
328
 
329
329
  rand = random.Random(rand_quantize_dropout_fixed_seed)
330
330
 
331
-
332
331
  rand_quantize_dropout_index = rand.randrange(self.quantize_dropout_cutoff_index, num_quant)
333
332
 
334
333
  if quant_dropout_multiple_of != 1:
@@ -496,7 +495,7 @@ class GroupedResidualVQ(Module):
496
495
  sample_codebook_temp = sample_codebook_temp,
497
496
  mask = mask,
498
497
  freeze_codebook = freeze_codebook,
499
- rand_quantize_dropout_fixed_seed = get_maybe_sync_seed(device)
498
+ rand_quantize_dropout_fixed_seed = get_maybe_sync_seed(device) if self.training else None
500
499
  )
501
500
 
502
501
  # invoke residual vq on each group