82 lines
1.7 KiB
JSON
82 lines
1.7 KiB
JSON
{
|
|
"creative": {
|
|
"ConfigName": "creative",
|
|
"max_length": 100,
|
|
"temperature": 0.9,
|
|
"top_k": 50,
|
|
"top_p": 0.9,
|
|
"do_sample": true,
|
|
"repetition_penalty": 1.2,
|
|
"no_repeat_ngram_size": 3,
|
|
"num_return_sequences": 1
|
|
},
|
|
"descriptive": {
|
|
"ConfigName": "detailed",
|
|
"max_length": 75,
|
|
"temperature": 0.7,
|
|
"top_k": 40,
|
|
"top_p": 0.85,
|
|
"do_sample": true,
|
|
"repetition_penalty": 1.3,
|
|
"no_repeat_ngram_size": 2,
|
|
"num_beams": 3
|
|
},
|
|
"technical": {
|
|
"ConfigName": "technical",
|
|
"max_length": 100,
|
|
"temperature": 0.3,
|
|
"top_k": 20,
|
|
"top_p": 0.7,
|
|
"do_sample": true,
|
|
"repetition_penalty": 1.4,
|
|
"no_repeat_ngram_size": 4,
|
|
"num_beams": 4
|
|
},
|
|
"diverse": {
|
|
"ConfigName": "diversed",
|
|
"max_length": 75,
|
|
"temperature": 1.2,
|
|
"top_k": 100,
|
|
"top_p": 0.95,
|
|
"do_sample": true,
|
|
"repetition_penalty": 1.5,
|
|
"no_repeat_ngram_size": 2,
|
|
"num_return_sequences": 1
|
|
},
|
|
"consistent": {
|
|
"ConfigName": "consistent",
|
|
"temperature": 0.3,
|
|
"num_beams": 4,
|
|
"repetition_penalty": 1.1,
|
|
"max_length": 50
|
|
},
|
|
"balanced": {
|
|
"ConfigName": "balanced",
|
|
"temperature": 0.7,
|
|
"top_k": 50,
|
|
"top_p": 0.9,
|
|
"num_beams": 4,
|
|
"repetition_penalty": 1.2
|
|
},
|
|
"photorealistic": {
|
|
"ConfigName": "photorealistic",
|
|
"max_length": 160,
|
|
"num_beams": 5,
|
|
"early_stopping": true,
|
|
"no_repeat_ngram_size": 3,
|
|
"num_return_sequences": 1,
|
|
"repetition_penalty": 1.3,
|
|
"length_penalty": 1.2
|
|
},
|
|
"refinement": {
|
|
"ConfigName": "refinements",
|
|
"max_length": 160,
|
|
"temperature": 0.6,
|
|
"top_k": 40,
|
|
"top_p": 0.8,
|
|
"do_sample": true,
|
|
"num_return_sequences": 1,
|
|
"no_repeat_ngram_size": 4,
|
|
"length_penalty": 1.1
|
|
}
|
|
} |