vector-quantize-pytorch 1.28.1__tar.gz → 1.28.2__tar.gz

{vector_quantize_pytorch-1.28.1 → vector_quantize_pytorch-1.28.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vector-quantize-pytorch
-Version: 1.28.1
+Version: 1.28.2
 Summary: Vector Quantization - Pytorch
 Project-URL: Homepage, https://pypi.org/project/vector-quantize-pytorch/
 Project-URL: Repository, https://github.com/lucidrains/vector-quantizer-pytorch

{vector_quantize_pytorch-1.28.1 → vector_quantize_pytorch-1.28.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "vector-quantize-pytorch"
-version = "1.28.1"
+version = "1.28.2"
 description = "Vector Quantization - Pytorch"
 authors = [
     { name = "Phil Wang", email = "lucidrains@gmail.com" }

{vector_quantize_pytorch-1.28.1 → vector_quantize_pytorch-1.28.2}/vector_quantize_pytorch/vector_quantize_pytorch.py

@@ -322,7 +322,7 @@ def directional_reparam(src, tgt, noise_variance = 5e-3):
     error_dir_norm = error_dir.norm(dim = -1, keepdim = True)
 
     noised_dir = error_dir + sqrt(noise_variance) * torch.randn_like(error_dir)
-    unit_noised_dir = l2norm(noised_dir)
+    unit_noised_dir = l2norm(noised_dir).detach()
 
     return src + unit_noised_dir * error_dir_norm
 
@@ -860,6 +860,7 @@ class VectorQuantize(Module):
         assert at_most_one_of(straight_through, rotation_trick, directional_reparam)
         self.rotation_trick = rotation_trick
 
+        assert not (directional_reparam and threshold_ema_dead_code == 0), 'periodic dead code replacement should be enabled when differential reparam method is turned on'
         self.directional_reparam = directional_reparam
         self.directional_reparam_variance = directional_reparam_variance