Skip to content

Commit 7a15d6b

Browse files
Update generative-proof-of-concept-CPU-preprocessing-in-memory.py
Try more settings for top_p top_k
1 parent 93e7284 commit 7a15d6b

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

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

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1069,8 +1069,12 @@ def complete_text(text):
10691069
do_sample=True,
10701070
max_new_tokens=20,
10711071
temperature=0.6,
1072-
top_k=40,
1073-
top_p=0.985,
1072+
# One set of recommendations
1073+
top_k=75,
1074+
top_p=0.95,
1075+
# Previous semi-working values
1076+
# top_k=40,
1077+
# top_p=0.985,
10741078
# repetition_penalty=1.2,
10751079
presence_penalty=1.2,
10761080
frequency_penalty=1.4

0 commit comments

Comments
 (0)