lightmare/whispercpp.py
1
1
Fork 1
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix wrong model key in __init__

Browse Source 
7be71f8fe7 changed the MODELS dict to download the models from HuggingFace. It also changed the keys used for the dict, but didn't change the way the keys are derived from the model name.
This commit is contained in:
Jan Beckmann 2022-12-26 14:04:58 +01:00 committed by GitHub
parent 275783f323
commit 6b93f9aa85
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
whispercpp.pyx
View File

@ -90,7 +90,7 @@ cdef class Whisper:
cdef whisper_full_params params
def __init__(self, model=DEFAULT_MODEL, pb=None):
model_fullname = f'model_ggml_{model}.bin'.encode('utf8')
model_fullname = f'ggml-{model}.bin'.encode('utf8')
download_model(model_fullname)
cdef bytes model_b = MODELS_DIR.encode('utf8') + b'/' + model_fullname
self.ctx = whisper_init(model_b)