From 149ba90104464d2b3cb9994925b5274b7cae0df3 Mon Sep 17 00:00:00 2001 From: Hans Date: Fri, 3 May 2024 18:09:18 +0800 Subject: [PATCH] test: disable GPU to fixed test result --- test/index.test.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/index.test.ts b/test/index.test.ts index 540fe32..097571b 100644 --- a/test/index.test.ts +++ b/test/index.test.ts @@ -4,7 +4,7 @@ import { loadModel } from '../lib' it('work fine', async () => { let tokens = '' - const model = await loadModel({ model: path.resolve(__dirname, './TinyLLama-v0.Q8_0.gguf') }) + const model = await loadModel({ model: path.resolve(__dirname, './TinyLLama-v0.Q8_0.gguf'), n_gpu_layers: 0 }) const result = await model.completion({ prompt: 'My name is Merve and my favorite', n_samples: 1,