test: disable GPU to fixed test result

mybigday · May 3, 2024 · 149ba90 · 149ba90
1 parent 3a4c2ef
commit 149ba90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,