Skip to content

Commit a49021a

Browse files
Update generative-proof-of-concept-CPU-preprocessing-in-memory.py
Add back separate frequency and presence penalties.
1 parent 2ac5dcb commit a49021a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1071,9 +1071,9 @@ def complete_text(text):
10711071
temperature=0.6,
10721072
top_k=40,
10731073
top_p=0.985,
1074-
repetition_penalty=1.2,
1075-
# presence_penalty=1.2,
1076-
# frequency_penalty=1.3
1074+
# repetition_penalty=1.2,
1075+
presence_penalty=1.2,
1076+
frequency_penalty=1.4
10771077
)
10781078

10791079
# Decode the result

0 commit comments

Comments
 (0)