happyz
/
gemma.cpp
mirror of https://github.com/google/gemma.cpp.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #98 from zeerd:patch-2 

PiperOrigin-RevId: 615769065
This commit is contained in:
Copybara-Service 2024-03-14 07:21:23 -07:00
parent 1408784d7b 46c1aca304
commit a2ef389897
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
gemma.cc
View File

@ -601,6 +601,9 @@ void GenerateImpl(GemmaImpl<TConfig>& gemma, size_t max_tokens,
} else if ((prompt.size() + max_generated_tokens) > max_tokens) { } else if ((prompt.size() + max_generated_tokens) > max_tokens) {
std::cout << "Warning: Prompt size + max_new_tokens exceeds max_tokens." std::cout << "Warning: Prompt size + max_new_tokens exceeds max_tokens."
<< std::endl; << std::endl;
} else if (pos >= max_tokens) {
std::cout << "Warning: pos exceeds max_tokens."
<< std::endl;
} }
} }