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

amd graphic card support

Browse Source
This commit is contained in:
Yaroslav P
2025-03-05 16:23:10 +02:00
parent 56a925463f
commit 049a168c81
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/_LocalTranscribe.py
View File

@@ -2,7 +2,7 @@ import os
import datetime import datetime
from glob import glob from glob import glob
import whisper import whisper
from torch import cuda, Generator from torch import backends, cuda, Generator
import colorama import colorama
from colorama import Back,Fore from colorama import Back,Fore
colorama.init(autoreset=True) colorama.init(autoreset=True)
@@ -41,10 +41,13 @@ def transcribe(path, glob_file, model=None, language=None, verbose=False):
""" """
# Check for GPU acceleration # Check for GPU acceleration
if cuda.is_available(): if backends.mps.is_available():
Generator('mps').manual_seed(42)
elif cuda.is_available():
Generator('cuda').manual_seed(42) Generator('cuda').manual_seed(42)
else: else:
Generator().manual_seed(42) Generator().manual_seed(42)
# Load model # Load model
model = whisper.load_model(model) model = whisper.load_model(model)
# Start main loop # Start main loop