Skip to content

Commit 6713887

Browse files
Update generative-proof-of-concept-CPU-preprocessing-in-memory.py
Try a top_p of 0.99.
1 parent 96deb86 commit 6713887

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

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

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1177,6 +1177,15 @@ def test_text(test_prompt: str, max_new_tokens: int, sample_number: int, result:
11771177
'repetition_penalty': None,
11781178
'presence_penalty': 1.4,
11791179
'frequency_penalty': 1.3
1180+
}, #
1181+
{
1182+
'max_new_tokens': max_new_tokens,
1183+
'temperature': 0.6,
1184+
'top_k': 75,
1185+
'top_p': 0.99,
1186+
'repetition_penalty': None,
1187+
'presence_penalty': 1.4,
1188+
'frequency_penalty': 1.4
11801189
}
11811190
]
11821191
# Default cases, no params

0 commit comments

Comments
 (0)