fix: increase max_new_tokens when testing generation models (#1244)
This commit is contained in:
@@ -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",
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user