Use YAML for presets and settings

This commit is contained in:
oobabooga 2023-05-28 22:34:12 -03:00
parent 2cf711f35e
commit 00ebea0b2a
42 changed files with 155 additions and 169 deletions

View file

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

View file

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

View file

@ -1 +0,0 @@
do_sample=False

View file

@ -0,0 +1 @@
do_sample: False

View file

@ -1,6 +0,0 @@
do_sample=True
top_p=0.5
top_k=40
temperature=0.7
repetition_penalty=1.2
typical_p=1.0

6
presets/Default.yaml Normal file
View file

@ -0,0 +1,6 @@
do_sample: true
top_p: 0.5
top_k: 40
temperature: 0.7
repetition_penalty: 1.2
typical_p: 1.0

View file

@ -1,6 +0,0 @@
do_sample=True
top_p=0.5
top_k=0
temperature=0.7
repetition_penalty=1.1
typical_p=0.19

View file

@ -0,0 +1,6 @@
do_sample: true
top_p: 0.5
top_k: 0
temperature: 0.7
repetition_penalty: 1.1
typical_p: 0.19

View file

@ -1,6 +0,0 @@
do_sample=True
top_p=1.0
top_k=0
temperature=0.66
repetition_penalty=1.1
typical_p=0.6

View file

@ -0,0 +1,6 @@
do_sample: true
top_p: 1.0
top_k: 0
temperature: 0.66
repetition_penalty: 1.1
typical_p: 0.6

View file

@ -1,6 +0,0 @@
do_sample=True
top_p=0.1
top_k=40
temperature=0.7
repetition_penalty=1.18
typical_p=1.0

View file

@ -0,0 +1,6 @@
do_sample: true
top_p: 0.1
top_k: 40
temperature: 0.7
repetition_penalty: 1.18
typical_p: 1.0

View file

@ -1,3 +0,0 @@
temperature=0.7
top_p=0.8
repetition_penalty=1.02

3
presets/MOSS.yaml Normal file
View file

@ -0,0 +1,3 @@
temperature: 0.7
top_p: 0.8
repetition_penalty: 1.02

View file

@ -1,4 +0,0 @@
do_sample=True
temperature=0.7
top_p=0.85
top_k=50

4
presets/Naive.yaml Normal file
View file

@ -0,0 +1,4 @@
do_sample: true
temperature: 0.7
top_p: 0.85
top_k: 50

View file

@ -1,6 +0,0 @@
do_sample=True
top_p=0.9
top_k=100
temperature=0.8
repetition_penalty=1.15
typical_p=1.0

View file

@ -0,0 +1,6 @@
do_sample: true
top_p: 0.9
top_k: 100
temperature: 0.8
repetition_penalty: 1.15
typical_p: 1.0

View file

@ -1,6 +0,0 @@
do_sample=True
top_p=1.0
top_k=100
temperature=2
repetition_penalty=1
typical_p=0.97

View file

@ -0,0 +1,6 @@
do_sample: true
top_p: 1.0
top_k: 100
temperature: 2
repetition_penalty: 1
typical_p: 0.97

View file

@ -1,6 +0,0 @@
do_sample=True
top_p=0.98
top_k=0
temperature=0.63
repetition_penalty=1.05
typical_p=1.0

View file

@ -0,0 +1,6 @@
do_sample: true
top_p: 0.98
top_k: 0
temperature: 0.63
repetition_penalty: 1.05
typical_p: 1.0

View file

@ -1,6 +0,0 @@
do_sample=True
top_p=0.85
top_k=12
temperature=2
repetition_penalty=1.15
typical_p=1.0

View file

@ -0,0 +1,6 @@
do_sample: true
top_p: 0.85
top_k: 12
temperature: 2
repetition_penalty: 1.15
typical_p: 1.0

View file

@ -1,6 +0,0 @@
do_sample=True
top_p=1.0
top_k=100
temperature=1.07
repetition_penalty=1.05
typical_p=1.0

View file

@ -0,0 +1,6 @@
do_sample: true
top_p: 1.0
top_k: 100
temperature: 1.07
repetition_penalty: 1.05
typical_p: 1.0

View file

@ -1,6 +0,0 @@
do_sample=True
top_p=1.0
top_k=0
temperature=0.44
repetition_penalty=1.15
typical_p=1.0

View file

@ -0,0 +1,6 @@
do_sample: true
top_p: 1.0
top_k: 0
temperature: 0.44
repetition_penalty: 1.15
typical_p: 1.0

View file

@ -1,6 +0,0 @@
do_sample=True
top_p=0.18
top_k=30
temperature=2.0
repetition_penalty=1.15
typical_p=1.0

View file

@ -0,0 +1,6 @@
do_sample: true
top_p: 0.18
top_k: 30
temperature: 2.0
repetition_penalty: 1.15
typical_p: 1.0

View file

@ -1,6 +0,0 @@
do_sample=True
top_p=0.73
top_k=0
temperature=0.72
repetition_penalty=1.1
typical_p=1.0

View file

@ -0,0 +1,6 @@
do_sample: true
top_p: 0.73
top_k: 0
temperature: 0.72
repetition_penalty: 1.1
typical_p: 1.0

View file

@ -1,9 +0,0 @@
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

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