mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-18 07:41:03 +00:00
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:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user