txya900619
commited on
Commit
·
2202dfb
1
Parent(s):
f634938
fix: use load_faster_whisper to load whisper
Browse files- app.py +1 -0
- 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}
|