kobim/whisper-local-transcribe
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add explicit device parameter to whisper.load_model()

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files

9
src/_LocalTranscribe.py
View File

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

BIN
src/__pycache__/_LocalTranscribe.cpython-312.pyc Normal file
View File

Binary file not shown.