putain quel con

Vermeille · Dec 12, 2024 · 6af25d1 · 6af25d1
1 parent 7e5c418
commit 6af25d1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchelie/nn/vq.py b/torchelie/nn/vq.py
@@ -89,7 +89,7 @@ def forward(
             quantized tensor, or (quantized tensor, indices) if
             `self.return_indices`
         """
-        if torch.is_floating_point(my_tensor):
+        if torch.is_floating_point(x):
             return self.quantize(x)
         else:
             return self.lookup(x)