fix: no llama on first gen

This commit is contained in:
linonetwo 2023-07-18 01:07:33 +08:00 committed by lin onetwo
parent d8c301c788
commit f86631b796

View file

@ -29,6 +29,7 @@ async function loadLLama(
...loadConfigOverwrite,
};
await llama.load(loadConfig);
return llama;
}
function unloadLLama() {
llama = undefined;
@ -43,8 +44,7 @@ function runLLama(
return new Observable<ILanguageModelWorkerResponse>((subscriber) => {
void (async function runLLamaObservableIIFE() {
if (llama === undefined) {
await loadLLama(loadConfig);
return;
llama = await loadLLama(loadConfig);
}
try {
let respondTimeout: NodeJS.Timeout | undefined;