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

Make gemma_test slightly more allowing on MultiTurn. 

PiperOrigin-RevId: 668097277
This commit is contained in:
Daniel Keysers 2024-08-27 12:39:41 -07:00 committed by Copybara-Service
parent 2308514e5a
commit 3c17911875
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
evals/gemma_test.cc
View File

@ -185,7 +185,10 @@ TEST_F(GemmaTest, Multiturn) {
dialog.clear(); dialog.clear();
model->Generate(runtime_config, tokens, abs_pos, s_env->MutableKVCache(), model->Generate(runtime_config, tokens, abs_pos, s_env->MutableKVCache(),
timing_info); timing_info);
EXPECT_TRUE(dialog.find("turquoise") != std::string::npos); // NOLINT fprintf(stderr, "decoded: %s\n", dialog.c_str());
bool remembered_turquoise = dialog.find("turquoise") != std::string::npos;
bool remembered_car = dialog.find("car") != std::string::npos;
EXPECT_TRUE(remembered_turquoise || remembered_car);
} }
static const char kJingleBells[] = R"( static const char kJingleBells[] = R"(