@@ -1186,6 +1186,15 @@ def test_text(test_prompt: str, max_new_tokens: int, sample_number: int, result:
11861186 'repetition_penalty' : None ,
11871187 'presence_penalty' : 1.4 ,
11881188 'frequency_penalty' : 1.4
1189+ },
1190+ {
1191+ 'max_new_tokens' : max_new_tokens ,
1192+ 'temperature' : 0.65 ,
1193+ 'top_k' : 75 ,
1194+ 'top_p' : 0.985 ,
1195+ 'repetition_penalty' : None ,
1196+ 'presence_penalty' : 1.4 ,
1197+ 'frequency_penalty' : 1.4
11891198 }
11901199 ]
11911200 # Default cases, no params
@@ -1222,10 +1231,10 @@ def test_text(test_prompt: str, max_new_tokens: int, sample_number: int, result:
12221231 # "And the earth was without form, and",
12231232 "And God said, Let there be light: and there " ,
12241233 # "Shall we all go to the river and"
1225- "Try to" ,
1234+ # "Try to",
12261235 # "You must go and",
12271236 "In the beginning God created the heavens" ,
1228- "The earth was formless and empty, with darkness over" ,
1237+ # "The earth was formless and empty, with darkness over",
12291238 # "God called the light 'day' and the darkness 'night,' marking evening and morning",
12301239 # "God called the expanse 'sky,' and there was",
12311240 # "The earth brought forth grass, seed-bearing"
0 commit comments