mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Rename additive_repetition_penalty to presence_penalty, add frequency_penalty (#4376)

Browse source
This commit is contained in:
tdrussell 2023-10-25 10:10:28 -05:00 • committed by GitHub
parent ef1489cd4d
commit 72f6fc6923
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
14 changed files with 64 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

23
modules/loaders.py
View file

@ -152,7 +152,8 @@ loaders_samplers = {
'tfs',
'top_a',
'repetition_penalty',
'additive_repetition_penalty',
'presence_penalty',
'frequency_penalty',
'repetition_penalty_range',
'encoder_repetition_penalty',
'no_repeat_ngram_size',
@ -186,7 +187,8 @@ loaders_samplers = {
'tfs',
'top_a',
'repetition_penalty',
'additive_repetition_penalty',
'presence_penalty',
'frequency_penalty',
'repetition_penalty_range',
'encoder_repetition_penalty',
'no_repeat_ngram_size',
@ -245,7 +247,8 @@ loaders_samplers = {
'tfs',
'top_a',
'repetition_penalty',
'additive_repetition_penalty',
'presence_penalty',
'frequency_penalty',
'repetition_penalty_range',
'encoder_repetition_penalty',
'no_repeat_ngram_size',
@ -275,7 +278,8 @@ loaders_samplers = {
'tfs',
'top_a',
'repetition_penalty',
'additive_repetition_penalty',
'presence_penalty',
'frequency_penalty',
'repetition_penalty_range',
'encoder_repetition_penalty',
'no_repeat_ngram_size',
@ -309,7 +313,8 @@ loaders_samplers = {
'tfs',
'top_a',
'repetition_penalty',
'additive_repetition_penalty',
'presence_penalty',
'frequency_penalty',
'repetition_penalty_range',
'encoder_repetition_penalty',
'no_repeat_ngram_size',
@ -339,6 +344,8 @@ loaders_samplers = {
'top_k',
'tfs',
'repetition_penalty',
'presence_penalty',
'frequency_penalty',
'mirostat_mode',
'mirostat_tau',
'mirostat_eta',
@ -357,7 +364,8 @@ loaders_samplers = {
'tfs',
'top_a',
'repetition_penalty',
'additive_repetition_penalty',
'presence_penalty',
'frequency_penalty',
'repetition_penalty_range',
'encoder_repetition_penalty',
'no_repeat_ngram_size',
@ -394,7 +402,8 @@ loaders_samplers = {
'tfs',
'top_a',
'repetition_penalty',
'additive_repetition_penalty',
'presence_penalty',
'frequency_penalty',
'repetition_penalty_range',
'encoder_repetition_penalty',
'no_repeat_ngram_size',