address #135

Author: lucidrains

pyproject.toml

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

vector_quantize_pytorch/residual_fsq.py

@@ -262,7 +262,7 @@ def forward(
 
         forward_kwargs = dict(
             return_all_codes = return_all_codes,
-            rand_quantize_dropout_fixed_seed = random.randint(0, 10_000_000)
+            rand_quantize_dropout_fixed_seed = random.randint(0, int(1e7))
         )
 
         # invoke residual vq on each group

vector_quantize_pytorch/residual_lfq.py

@@ -252,7 +252,7 @@ def forward(
         forward_kwargs = dict(
             mask = mask,
             return_all_codes = return_all_codes,
-            rand_quantize_dropout_fixed_seed = random.randint(0, 1e7)
+            rand_quantize_dropout_fixed_seed = random.randint(0, int(1e7))
         )
 
         # invoke residual vq on each group

vector_quantize_pytorch/residual_vq.py

@@ -288,7 +288,7 @@ def forward(
             sample_codebook_temp = sample_codebook_temp,
             mask = mask,
             freeze_codebook = freeze_codebook,
-            rand_quantize_dropout_fixed_seed = random.randint(0, 1e7)
+            rand_quantize_dropout_fixed_seed = random.randint(0, int(1e7))
         )
 
         # invoke residual vq on each group