Update modeling_wavcoch.py

modeling_wavcoch.py

@@ -508,7 +508,7 @@ class WavCoch(PreTrainedModel):
             if pad:
                 wav = F.pad(wav, (self.N - self.hop_length, 0), mode="constant", value=0.0)
 
-            codes = self.quantize(wav, pad=False)  # already padded
+            codes = self.quantize(wav, pad=False).long()  # already padded
             return BatchEncoding({"input_values": codes, "input_ids": codes})
 
         # Training / reconstruction mode