Explorar o código

improved warning message for English-only models

Jong Wook Kim %!s(int64=2) %!d(string=hai) anos
pai
achega
2d3032de01
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      whisper/transcribe.py

+ 1 - 1
whisper/transcribe.py

@@ -257,7 +257,7 @@ def cli():
     device: str = args.pop("device")
     os.makedirs(output_dir, exist_ok=True)
 
-    if model_name.endswith(".en") and args["language"] != "en":
+    if model_name.endswith(".en") and args["language"] not in {"en", "English"}:
         warnings.warn(f"{model_name} is an English-only model but receipted '{args['language']}'; using English instead.")
         args["language"] = "en"