vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add missing change...

Browse source
This commit is contained in:
goerch 2023-09-14 17:30:53 +02:00
parent 01b0105890
commit c7c0fcbfe0
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
examples/train-text-from-scratch/train-text-from-scratch.cpp
View file

@ -965,10 +965,10 @@ int tokenize_file(struct llama_context * lctx, const char * filename, std::vecto
buf[size] = '\0'; buf[size] = '\0';
int n_tokens = llama_tokenize(lctx, buf.data(), out.data(), out.size(), false); int n_tokens = llama_tokenize(lctx, buf.data(), buf.size(), out.data(), out.size(), false);
if (n_tokens < 0) { if (n_tokens < 0) {
out.resize(-n_tokens); out.resize(-n_tokens);
n_tokens = llama_tokenize(lctx, buf.data(), out.data(), out.size(), false); n_tokens = llama_tokenize(lctx, buf.data(), buf.size(), out.data(), out.size(), false);
} }
GGML_ASSERT(n_tokens >= 0); GGML_ASSERT(n_tokens >= 0);
out.resize(n_tokens); out.resize(n_tokens);