vector-quantize-pytorch 1.17.1__tar.gz → 1.17.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 (26) hide show
  1. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/PKG-INFO +1 -1
  2. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/pyproject.toml +1 -1
  3. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/vector_quantize_pytorch/lookup_free_quantization.py +3 -1
  4. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/.github/workflows/build.yml +0 -0
  5. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/.github/workflows/python-publish.yml +0 -0
  6. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/.github/workflows/test.yml +0 -0
  7. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/.gitignore +0 -0
  8. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/LICENSE +0 -0
  9. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/README.md +0 -0
  10. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/examples/autoencoder.py +0 -0
  11. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/examples/autoencoder_fsq.py +0 -0
  12. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/examples/autoencoder_lfq.py +0 -0
  13. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/images/fsq.png +0 -0
  14. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/images/lfq.png +0 -0
  15. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/images/vq.png +0 -0
  16. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/ruff.toml +0 -0
  17. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/tests/test_latent_quantization.py +0 -0
  18. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/tests/test_readme.py +0 -0
  19. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/vector_quantize_pytorch/__init__.py +0 -0
  20. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/vector_quantize_pytorch/finite_scalar_quantization.py +0 -0
  21. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/vector_quantize_pytorch/latent_quantization.py +0 -0
  22. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/vector_quantize_pytorch/random_projection_quantizer.py +0 -0
  23. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/vector_quantize_pytorch/residual_fsq.py +0 -0
  24. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/vector_quantize_pytorch/residual_lfq.py +0 -0
  25. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/vector_quantize_pytorch/residual_vq.py +0 -0
  26. {vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/vector_quantize_pytorch/vector_quantize_pytorch.py +0 -0
{vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: vector-quantize-pytorch
3
- Version: 1.17.1
3
+ Version: 1.17.2
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.17.1 → vector_quantize_pytorch-1.17.2}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "vector-quantize-pytorch"
3
- version = "1.17.1"
3
+ version = "1.17.2"
4
4
  description = "Vector Quantization - Pytorch"
5
5
  authors = [
6
6
  { name = "Phil Wang", email = "lucidrains@gmail.com" }
{vector_quantize_pytorch-1.17.1 → vector_quantize_pytorch-1.17.2}/vector_quantize_pytorch/lookup_free_quantization.py RENAMED
@@ -10,7 +10,9 @@ from math import log2, ceil
10
10
  from functools import partial, cache
11
11
  from collections import namedtuple
12
12
  from contextlib import nullcontext
13
+
13
14
  import torch.distributed as dist
15
+ from torch.distributed import nn as dist_nn
14
16
 
15
17
  import torch
16
18
  from torch import nn, einsum
@@ -36,7 +38,7 @@ def maybe_distributed_mean(t):
36
38
  if not is_distributed():
37
39
  return t
38
40
 
39
- dist.nn.all_reduce(t)
41
+ dist_nn.all_reduce(t)
40
42
  t = t / dist.get_world_size()
41
43
  return t
42
44