Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-11-24 06:26:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add capability to specify ffmpeg path

Browse Source
This commit is contained in:
Akande peter 2025-01-27 12:56:49 +01:00
parent 517a43ecd1
commit 197922dcc0
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
whisper/__init__.py
View File

@ -50,6 +50,8 @@ _ALIGNMENT_HEADS = {
"turbo": b"ABzY8j^C+e0{>%RARaKHP%t(lGR*)0g!tONPyhe`", "turbo": b"ABzY8j^C+e0{>%RARaKHP%t(lGR*)0g!tONPyhe`",
} }
ffmpeg_path: str = "ffmpeg"
def _download(url: str, root: str, in_memory: bool) -> Union[bytes, str]: def _download(url: str, root: str, in_memory: bool) -> Union[bytes, str]:
os.makedirs(root, exist_ok=True) os.makedirs(root, exist_ok=True)

3
whisper/audio.py
View File

@ -7,6 +7,7 @@ import numpy as np
import torch import torch
import torch.nn.functional as F import torch.nn.functional as F
import whisper
from .utils import exact_div from .utils import exact_div
# hard-coded audio hyperparameters # hard-coded audio hyperparameters
@ -43,7 +44,7 @@ def load_audio(file: str, sr: int = SAMPLE_RATE):
# and resampling as necessary. Requires the ffmpeg CLI in PATH. # and resampling as necessary. Requires the ffmpeg CLI in PATH.
# fmt: off # fmt: off
cmd = [ cmd = [
"ffmpeg", whisper.ffmpeg_path,
"-nostdin", "-nostdin",
"-threads", "0", "-threads", "0",
"-i", file, "-i", file,