From 6b93f9aa8501790b2ed5ee3186c227c70bf343f2 Mon Sep 17 00:00:00 2001 From: Jan Beckmann Date: Mon, 26 Dec 2022 14:04:58 +0100 Subject: [PATCH] Fix wrong model key in __init__ 7be71f8fe7135064a9c75cc93e2164733e9867f1 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. --- whispercpp.pyx | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/whispercpp.pyx b/whispercpp.pyx index d1fc5d7..486e1fc 100644 --- a/whispercpp.pyx +++ b/whispercpp.pyx @@ -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)