Skip to content

Commit

Permalink
Revert "test: change test model to bypass OP problem"
Browse files Browse the repository at this point in the history
This reverts commit 478cc73.
  • Loading branch information
hans00 committed May 3, 2024
1 parent 149ba90 commit e507eae
Show file tree
Hide file tree
Showing 4 changed files with 2 additions and 2 deletions.
Binary file removed test/TinyLLama-v0.Q8_0.gguf
Binary file not shown.
2 changes: 1 addition & 1 deletion test/__snapshots__/index.test.ts.snap
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

exports[`work fine 1`] = `
{
"text": " day. He was so excited to play with his",
"text": " swochadoorter scientific WindowsCa occupiedrå alta",
"tokens_evaluated": 18,
"tokens_predicted": 10,
"truncated": false,
Expand Down
2 changes: 1 addition & 1 deletion test/index.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -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'), n_gpu_layers: 0 })
const model = await loadModel({ model: path.resolve(__dirname, './tiny-random-llama.gguf'), n_gpu_layers: 0 })
const result = await model.completion({
prompt: 'My name is Merve and my favorite',
n_samples: 1,
Expand Down
Binary file added test/tiny-random-llama.gguf
Binary file not shown.

0 comments on commit e507eae

Please sign in to comment.