Fix localization bug

jhj0517 · Oct 20, 2024 · 84fd983 · 84fd983
1 parent f7e3728
commit 84fd983
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/whisper/whisper_parameter.py b/modules/whisper/whisper_parameter.py
@@ -308,7 +308,7 @@ def to_yaml(self) -> Dict:
         data = {
             "whisper": {
                 "model_size": self.model_size,
-                "lang": AUTOMATIC_DETECTION if self.lang is None else self.lang,
+                "lang": AUTOMATIC_DETECTION.unwrap() if self.lang is None else self.lang,
                 "is_translate": self.is_translate,
                 "beam_size": self.beam_size,
                 "log_prob_threshold": self.log_prob_threshold,