Add Contrastive Search preset #197

2023-03-09 10:27:11 -03:00
parent 557c773df7
commit e7adf5fe4e
3 changed files with 3 additions and 6 deletions
--- a/presets/Contrastive
+++ b/presets/Contrastive
@@ -0,0 +1,3 @@
+do_sample=False
+penalty_alpha=0.6
+top_k=4
--- a/presets/Instruct
+++ b/presets/Instruct
@@ -1,5 +0,0 @@
-top_p=0.95
-temperature=0.5
-penalty_alpha=0.6
-top_k=4
-repetition_penalty=1.03