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
def __init__(self, char* model=DEFAULT_MODEL, pb=None):
model_fullname = f'model_ggml_{model.decode()}.bin'
download_model(model_fullname)
self.ctx = whisper_init(model_fullname)
model = f'model_ggml_{model.decode()}.bin'.encode('utf8')
download_model(model)
self.ctx = whisper_init(model)
self.params = default_params()
def __dealloc__(self):