yet another fix for residual fsq

Author: lucidrains

setup.py

@@ -3,7 +3,7 @@
 setup(
   name = 'vector_quantize_pytorch',
   packages = find_packages(),
-  version = '1.11.1',
+  version = '1.11.2',
   license='MIT',
   description = 'Vector Quantization - Pytorch',
   long_description_content_type = 'text/markdown',

vector_quantize_pytorch/residual_fsq.py

@@ -163,7 +163,6 @@ def forward(
 
                 if should_quantize_dropout and quantizer_index > rand_quantize_dropout_index:
                     all_indices.append(null_indices)
-                    all_losses.append(null_loss)
                     continue
 
                 quantized, indices = layer(residual / scale)
@@ -178,7 +177,7 @@ def forward(
 
         quantized_out = self.project_out(quantized_out)
 
-        # stack all losses and indices
+        # stack all indices
 
         all_indices = torch.stack(all_indices, dim = -1)