Reorganize the generation parameters

This commit is contained in:
oobabooga 2023-02-17 16:18:01 -03:00
parent 3923ac967f
commit 6b9ac2f88e
4 changed files with 19 additions and 19 deletions

View file

@ -0,0 +1,9 @@
num_beams=10
min_length=200
length_penalty =1.4
no_repeat_ngram_size=2
early_stopping=True
temperature=0.7
top_k=150
top_p=0.92
repetition_penalty=4.5