Skip to content

Commit

Permalink
Update README.md repetition penalty=1.1
Browse files Browse the repository at this point in the history
  • Loading branch information
Saibo-creator authored Feb 24, 2024
1 parent 42abdbe commit 81f6f53
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,7 @@ if __name__ == "__main__":
max_length=50,
num_beams=2,
logits_processor=[grammar_processor],
repetition_penalty=5.0,
repetition_penalty=1.1,
num_return_sequences=1,
)
# decode output
Expand Down

0 comments on commit 81f6f53

Please sign in to comment.