Skip to content

Commit 2d3032d

Browse files
committed
improved warning message for English-only models
1 parent 8cf36f3 commit 2d3032d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

whisper/transcribe.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -257,7 +257,7 @@ def cli():
257257
device: str = args.pop("device")
258258
os.makedirs(output_dir, exist_ok=True)
259259

260-
if model_name.endswith(".en") and args["language"] != "en":
260+
if model_name.endswith(".en") and args["language"] not in {"en", "English"}:
261261
warnings.warn(f"{model_name} is an English-only model but receipted '{args['language']}'; using English instead.")
262262
args["language"] = "en"
263263

0 commit comments

Comments
 (0)