vector-quantize-pytorch 1.23.2__tar.gz → 1.23.3__tar.gz

{vector_quantize_pytorch-1.23.2 → vector_quantize_pytorch-1.23.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vector-quantize-pytorch
-Version: 1.23.2
+Version: 1.23.3
 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.23.2 → vector_quantize_pytorch-1.23.3}/pyproject.toml

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

{vector_quantize_pytorch-1.23.2 → vector_quantize_pytorch-1.23.3}/vector_quantize_pytorch/vector_quantize_pytorch.py

@@ -1263,7 +1263,7 @@ class VectorQuantize(Module):
             # calculate codebook diversity loss (negative of entropy) if needed
 
             if self.has_codebook_diversity_loss:
-                prob = (-distances * self.codebook_diversity_temperature).softmax(dim = -1)
+                prob = (distances * self.codebook_diversity_temperature).softmax(dim = -1)
                 avg_prob = reduce(prob, '... n l -> n l', 'mean')
                 codebook_diversity_loss = -entropy(avg_prob).mean()