fix: Remove labels kwarg from Whisper tokenizer padding

alexandrainst · Dec 13, 2023 · 6b80161 · 6b80161
1 parent 6430fd7
commit 6b80161
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/coral_models/whisper.py b/src/coral_models/whisper.py
@@ -92,7 +92,7 @@ def torch_call(self, features: list[dict]) -> BatchFeature:
 
         # Pad the labels to max length
         labels_batch = self.processor.tokenizer.pad(
-            labels=label_features,
+            label_features,
             padding=self.padding,
             return_tensors=self.return_tensors,
             max_length=512,