From 0447f37fb744a1b350f6b92e4d140dbdb1c8d3ec Mon Sep 17 00:00:00 2001 From: yum Date: Mon, 11 Sep 2023 13:46:57 -0700 Subject: Pin huggingface_hub to 0.16.4 0.17.x are breaking faster_whisper's ability to download models. Also: * Start using frozen requirements.txt. * Conditionally install torch & legacy whisper only when doing mechanical optimization. --- Scripts/transcribe_v2.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) (limited to 'Scripts/transcribe_v2.py') diff --git a/Scripts/transcribe_v2.py b/Scripts/transcribe_v2.py index 6377ff4..9eed78f 100644 --- a/Scripts/transcribe_v2.py +++ b/Scripts/transcribe_v2.py @@ -4,9 +4,7 @@ from faster_whisper import WhisperModel from functools import partial from profanity_filter import ProfanityFilter from pydub import AudioSegment -from scipy.optimize import minimize from sentence_splitter import split_text_into_sentences -from whisper.normalizers import EnglishTextNormalizer import app_config import argparse @@ -814,6 +812,12 @@ def evaluate(cfg, def optimize(cfg, experiments: typing.List[typing.Tuple[str, str]]): + install_in_venv(["git+https://github.com/openai/whisper.git", + "scipy"]) + + from scipy.optimize import minimize + from whisper.normalizers import EnglishTextNormalizer + def wrapper_to_optimize(x): s = 0 for audio_path, control_path in experiments: -- cgit v1.2.3