Add explicit device parameter to whisper.load_model()

Co-authored-by: soderstromkr <23003509+soderstromkr@users.noreply.github.com>
This commit is contained in:
copilot-swe-agent[bot]
2026-01-22 13:00:21 +00:00
parent 953c71ab28
commit 0dab0d9bea
2 changed files with 6 additions and 3 deletions

View File

@@ -40,16 +40,19 @@ def transcribe(path, glob_file, model=None, language=None, verbose=False):
within the specified path.
"""
# Check for GPU acceleration
# Check for GPU acceleration and set device
if backends.mps.is_available():
device = 'mps'
Generator('mps').manual_seed(42)
elif cuda.is_available():
device = 'cuda'
Generator('cuda').manual_seed(42)
else:
device = 'cpu'
Generator().manual_seed(42)
# Load model
model = whisper.load_model(model)
# Load model on the correct device
model = whisper.load_model(model, device=device)
# Start main loop
files_transcripted=[]
for file in glob_file:

Binary file not shown.