diff --git a/test/TinyLLama-v0.Q8_0.gguf b/test/TinyLLama-v0.Q8_0.gguf new file mode 100644 index 0000000..0d22107 Binary files /dev/null and b/test/TinyLLama-v0.Q8_0.gguf differ diff --git a/test/__snapshots__/index.test.ts.snap b/test/__snapshots__/index.test.ts.snap index 997c35c..2f4b13d 100644 --- a/test/__snapshots__/index.test.ts.snap +++ b/test/__snapshots__/index.test.ts.snap @@ -2,7 +2,7 @@ exports[`work fine 1`] = ` { - "text": " swochadoorter scientific WindowsCa occupiedrå alta", + "text": " day. He was so excited to play with his", "tokens_evaluated": 18, "tokens_predicted": 10, "truncated": false, diff --git a/test/index.test.ts b/test/index.test.ts index e27ee26..540fe32 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, './tiny-random-llama.gguf') }) + const model = await loadModel({ model: path.resolve(__dirname, './TinyLLama-v0.Q8_0.gguf') }) const result = await model.completion({ prompt: 'My name is Merve and my favorite', n_samples: 1, diff --git a/test/tiny-random-llama.gguf b/test/tiny-random-llama.gguf deleted file mode 100644 index 55b2b40..0000000 Binary files a/test/tiny-random-llama.gguf and /dev/null differ