txya900619 commited on
Commit
2202dfb
·
1 Parent(s): f634938

fix: use load_faster_whisper to load whisper

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. configs/models.yaml +1 -1
app.py CHANGED
@@ -29,6 +29,7 @@ def load_faster_whisper(model_id: str):
29
 
30
 
31
  OmegaConf.register_new_resolver("load_model", load_model)
 
32
 
33
  models_config = OmegaConf.to_object(OmegaConf.load("configs/models.yaml"))
34
 
 
29
 
30
 
31
  OmegaConf.register_new_resolver("load_model", load_model)
32
+ OmegaConf.register_new_resolver("load_faster_whisper", load_faster_whisper)
33
 
34
  models_config = OmegaConf.to_object(OmegaConf.load("configs/models.yaml"))
35
 
configs/models.yaml CHANGED
@@ -1,4 +1,4 @@
1
  whisper-large-v3-finetune:
2
- model: kenkone/kenkone-whisper-large-v3-ct2
3
  parakeet-0.6b-custom-tokenizer:
4
  model: ${load_model:kenkone/parakeet-ctc-0.6b-custom-tokenizer}
 
1
  whisper-large-v3-finetune:
2
+ model: ${load_faster_whisper:kenkone/kenkone-whisper-large-v3-ct2}
3
  parakeet-0.6b-custom-tokenizer:
4
  model: ${load_model:kenkone/parakeet-ctc-0.6b-custom-tokenizer}