diff options
| author | yum <yum.food.vr@gmail.com> | 2023-09-11 13:46:57 -0700 |
|---|---|---|
| committer | yum <yum.food.vr@gmail.com> | 2023-09-11 13:53:14 -0700 |
| commit | 0447f37fb744a1b350f6b92e4d140dbdb1c8d3ec (patch) | |
| tree | df57c1fd17bc8a9533713afb1663c17f6df8cf24 /Scripts/transcribe_v2.py | |
| parent | d3c325c4c4dd954a75267b013f33f5f3c5d041bc (diff) | |
Pin huggingface_hub to 0.16.4v0.15.2
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.
Diffstat (limited to 'Scripts/transcribe_v2.py')
| -rw-r--r-- | Scripts/transcribe_v2.py | 8 |
1 files changed, 6 insertions, 2 deletions
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: |
