victor-upmeet / whisperx-replicate

19 stars 28 forks source link

Fix to HFValidationError #6

Open sonicbaume opened 3 months ago

sonicbaume commented 3 months ago

The following error occurs on the latest version:

Traceback (most recent call last):
File "/root/.pyenv/versions/3.11.9/lib/python3.11/site-packages/cog/server/worker.py", line 221, in _predict
result = predict(**payload)
^^^^^^^^^^^^^^^^^^
File "/src/predict.py", line 132, in predict
model = whisperx.load_model(whisper_arch, device, compute_type=compute_type, language=language,
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/root/.pyenv/versions/3.11.9/lib/python3.11/site-packages/whisperx/asr.py", line 288, in load_model
model = model or WhisperModel(whisper_arch,
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/root/.pyenv/versions/3.11.9/lib/python3.11/site-packages/faster_whisper/transcribe.py", line 127, in __init__
model_path = download_model(
^^^^^^^^^^^^^^^
File "/root/.pyenv/versions/3.11.9/lib/python3.11/site-packages/faster_whisper/utils.py", line 103, in download_model
return huggingface_hub.snapshot_download(repo_id, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/root/.pyenv/versions/3.11.9/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 106, in _inner_fn
validate_repo_id(arg_value)
File "/root/.pyenv/versions/3.11.9/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 154, in validate_repo_id
raise HFValidationError(
huggingface_hub.errors.HFValidationError: Repo id must be in the form 'repo_name' or 'namespace/repo_name': './models/faster-whisper-large-v3'. Use `repo_type` argument if needed.

This change sets whisper_arch to large-v3, which I believe still loads faster-whisper, and fixes the error.