Skip to content

Commit

Permalink
test: swap llama out for a mini model
Browse files Browse the repository at this point in the history
  • Loading branch information
Rexwang8 committed Oct 17, 2024
1 parent 1744c2c commit e184561
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion gpt_bpe_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -1471,7 +1471,7 @@ func TestModelDownloadLlama(t *testing.T) {
// the vocab.json and merges.txt files are stored in the
// tokenizer.json file. We want to check if we are able to
// download the model and extract the vocab.json and merges.txt
modelId := "georgesung/llama2_7b_chat_uncensored"
modelId := "Maykeye/TinyLLama-v0"
destPath := "./TestModelDownloadLlama"
err := downloadModel(modelId, destPath)
if err != nil {
Expand Down

0 comments on commit e184561

Please sign in to comment.