seed-vc: Use openai/whisper-small (#61)

Use openai/whisper-small as aa a workaround for whisper-large-v3 being unavailable and newer whisper versions being incompatible as a hotfix
This commit is contained in:
Madhukar Mishra
2024-12-04 18:24:59 +05:30
committed by GitHub
parent 8785206e14
commit 46302b7c0c
+1 -1
View File
@@ -110,7 +110,7 @@ def load_models(args):
whisper_name = (
model_params.speech_tokenizer.whisper_name
if hasattr(model_params.speech_tokenizer, "whisper_name")
else "whisper-large-v3"
else "openai/whisper-small"
)
whisper_model = WhisperModel.from_pretrained(
whisper_name, torch_dtype=torch.float16