This commit is contained in:
Luke Southam 2022-12-11 01:49:59 +00:00
parent af83eab0b5
commit 92fb3d452f

View File

@ -83,9 +83,9 @@ cdef class Whisper:
cdef whisper_full_params params cdef whisper_full_params params
def __init__(self, char* model=DEFAULT_MODEL, pb=None): def __init__(self, char* model=DEFAULT_MODEL, pb=None):
model_fullname = f'model_ggml_{model.decode()}.bin' model = f'model_ggml_{model.decode()}.bin'.encode('utf8')
download_model(model_fullname) download_model(model)
self.ctx = whisper_init(model_fullname) self.ctx = whisper_init(model)
self.params = default_params() self.params = default_params()
def __dealloc__(self): def __dealloc__(self):