fix(model): drop language parameter in get_decoder_prompt_ids method (#…

…201)
KevKibe · Sep 24, 2024 · 51335fa · 51335fa
1 parent 6f983ea
commit 51335fa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/training/whisper_model_prep.py b/src/training/whisper_model_prep.py
@@ -126,7 +126,7 @@ def initialize_model(self) -> WhisperForConditionalGeneration:
                 self.model_id,
                 low_cpu_mem_usage = True
             )
-            model.config.forced_decoder_ids = processor.get_decoder_prompt_ids(language=self.language, task=self.processing_task)
+            model.config.forced_decoder_ids = processor.get_decoder_prompt_ids(task=self.processing_task)
             # model.config.suppress_tokens = []
             model.config.use_cache = True
             model.generation_config.language = self.language if self.processing_task == "transcribe" else "en"