Skip to content

Commit 7b4120a

Browse files
Update generative-proof-of-concept-CPU-preprocessing-in-memory.py
Try top_p 0.85 temperature 0.65, cut underperforming prompts.
1 parent 6713887 commit 7b4120a

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

generative-proof-of-concept-CPU-preprocessing-in-memory.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)