This commit is contained in:
Josh Yan 2024-08-28 11:07:43 -07:00
parent 340162fbc3
commit 01b20fe6f1

View File

@ -41,7 +41,7 @@ func TestEstimateGPULayers(t *testing.T) {
"tokenizer.ggml.tokens": []string{" "}, "tokenizer.ggml.tokens": []string{" "},
"tokenizer.ggml.scores": []float32{0}, "tokenizer.ggml.scores": []float32{0},
"tokenizer.ggml.token_type": []int32{0}, "tokenizer.ggml.token_type": []int32{0},
}, tensors, func(api.ProgressResponse){}) }, tensors, func(api.ProgressResponse) {})
require.NoError(t, err) require.NoError(t, err)
ggml, err := LoadModel(f.Name(), 0) ggml, err := LoadModel(f.Name(), 0)