Add Contrastive Search preset #197

This commit is contained in:
oobabooga 2023-03-09 10:27:11 -03:00
parent 557c773df7
commit e7adf5fe4e
3 changed files with 3 additions and 6 deletions

View file

@ -0,0 +1,3 @@
do_sample=False
penalty_alpha=0.6
top_k=4

View file

@ -1,5 +0,0 @@
top_p=0.95
temperature=0.5
penalty_alpha=0.6
top_k=4
repetition_penalty=1.03