fix: increase max_new_tokens when testing generation models (#1244)

This commit is contained in:
Yineng Zhang
2024-08-28 22:12:36 +10:00
committed by GitHub
parent 6c49831394
commit 66975360e7
2 changed files with 6 additions and 6 deletions

View File

@@ -30,7 +30,7 @@ DEFAULT_PROMPTS = [
# the output of gemma-2-2b from SRT is unstable on the commented prompt
# "The capital of France is",
"Apple is red. Banana is Yellow. " * 800 + "Apple is",
"The capital of the United Kindom is",
"The capital of the United Kingdom is",
"Today is a sunny day and I like",
"AI is a field of computer science focused on",
]