Update moshi/moshi/quantization/core_vq.py

Co-authored-by: Ishita Mediratta <[email protected]>
kyutai-labs · Oct 14, 2024 · 7bc4362 · 7bc4362
1 parent c26dfc7
commit 7bc4362
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/moshi/moshi/quantization/core_vq.py b/moshi/moshi/quantization/core_vq.py
@@ -412,7 +412,6 @@ def forward(self, x: torch.Tensor, initialize: bool = True) -> _VQForwardResult:
         else:
             loss = zero_scalar(x.device)
 
-        loss = zero_scalar(x.device)
 
         quantized = self.project_out(quantized)
         quantized = self._rearrange_output(quantized)