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

Fix test for 2b - update prompt 

PiperOrigin-RevId: 667878053
This commit is contained in:
Jan Wassenberg 2024-08-27 00:56:14 -07:00 committed by Copybara-Service
parent 48d0801fb0
commit c4303cd89b
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
evals/gemma_test.cc
View File

@ -172,12 +172,12 @@ TEST_F(GemmaTest, Multiturn) {
}; };
TimingInfo timing_info{.verbosity = 0}; TimingInfo timing_info{.verbosity = 0};
// First "say" something slightly unusual. // First "say" something slightly unusual.
std::string mutable_prompt = "The color of my car is turquoise."; std::string mutable_prompt = "I have a car and its color is turquoise.";
std::vector<int> tokens = WrapAndTokenize(model->Tokenizer(), model->Info(), std::vector<int> tokens = WrapAndTokenize(model->Tokenizer(), model->Info(),
abs_pos, mutable_prompt); abs_pos, mutable_prompt);
model->Generate(runtime_config, tokens, abs_pos, s_env->MutableKVCache(), model->Generate(runtime_config, tokens, abs_pos, s_env->MutableKVCache(),
timing_info); timing_info);
mutable_prompt = "Can you repeat to me what I just said?"; mutable_prompt = "Please repeat all prior statements.";
tokens = WrapAndTokenize(model->Tokenizer(), model->Info(), abs_pos, tokens = WrapAndTokenize(model->Tokenizer(), model->Info(), abs_pos,
mutable_prompt); mutable_prompt);
// Reset the `dialog` string here, then check that the model actually has // Reset the `dialog` string here, then check that the model actually has