diff --git a/whisper/transcribe.py b/whisper/transcribe.py index 0ee42a8..7000cfa 100644 --- a/whisper/transcribe.py +++ b/whisper/transcribe.py @@ -273,7 +273,7 @@ def cli(): temperature = [temperature] from . import load_model - model = load_model(model_name).to(device) + model = load_model(model_name, device=device) for audio_path in args.pop("audio"): result = transcribe(