mirror of
https://github.com/SillyTavern/SillyTavern.git
synced 2025-06-05 21:59:27 +02:00
Fix default NAI presets. Add NAI bad words
This commit is contained in:
@ -1,13 +1,17 @@
|
|||||||
{
|
{
|
||||||
"order": [3, 0],
|
"order": [2, 1, 3, 0],
|
||||||
"temperature": 1.11,
|
"temperature": 0.63,
|
||||||
"max_length": 90,
|
"max_length": 90,
|
||||||
"min_length": 1,
|
"min_length": 1,
|
||||||
"tail_free_sampling": 0.68,
|
"tail_free_sampling": 0.975,
|
||||||
"repetition_penalty": 1.11,
|
"repetition_penalty": 1.148125,
|
||||||
"repetition_penalty_range": 320,
|
"repetition_penalty_range": 2048,
|
||||||
"repetition_penalty_frequency": 0,
|
"repetition_penalty_frequency": 0,
|
||||||
"repetition_penalty_presence": 0,
|
"repetition_penalty_presence": 0,
|
||||||
"repetition_penalty_slope": 0,
|
"repetition_penalty_slope": 0.09,
|
||||||
"max_context":2048
|
"max_context":2048,
|
||||||
}
|
"top_p": 0.975,
|
||||||
|
"top_k": 0,
|
||||||
|
"top_a": 1,
|
||||||
|
"typical_p": 1
|
||||||
|
}
|
||||||
|
@ -1,13 +1,17 @@
|
|||||||
{
|
{
|
||||||
"order": [3, 0],
|
"order": [3, 4, 5, 2, 0],
|
||||||
"temperature": 1.7,
|
"temperature": 1.33,
|
||||||
"max_length": 90,
|
"max_length": 90,
|
||||||
"min_length": 1,
|
"min_length": 1,
|
||||||
"tail_free_sampling": 0.66,
|
"tail_free_sampling": 0.937,
|
||||||
"repetition_penalty": 1.06,
|
"repetition_penalty": 1.05,
|
||||||
"repetition_penalty_range": 340,
|
"repetition_penalty_range": 560,
|
||||||
"repetition_penalty_frequency": 0,
|
"repetition_penalty_frequency": 0,
|
||||||
"repetition_penalty_presence": 0,
|
"repetition_penalty_presence": 0,
|
||||||
"repetition_penalty_slope": 0,
|
"repetition_penalty_slope": 0.18,
|
||||||
"max_context": 2048
|
"max_context": 2048,
|
||||||
}
|
"top_p": 0.88,
|
||||||
|
"top_k": 0,
|
||||||
|
"top_a": 0.085,
|
||||||
|
"typical_p": 0.985
|
||||||
|
}
|
||||||
|
@ -154,9 +154,9 @@ export function getNovelGenerationData(finalPromt, this_settings, this_amount_ge
|
|||||||
"typical_p": this_settings.typical_p,
|
"typical_p": this_settings.typical_p,
|
||||||
//"stop_sequences": {{187}},
|
//"stop_sequences": {{187}},
|
||||||
//bad_words_ids = {{50256}, {0}, {1}};
|
//bad_words_ids = {{50256}, {0}, {1}};
|
||||||
//generate_until_sentence = true;
|
"generate_until_sentence": true,
|
||||||
"use_cache": false,
|
"use_cache": false,
|
||||||
//use_string = true;
|
"use_string": true,
|
||||||
"return_full_text": false,
|
"return_full_text": false,
|
||||||
"prefix": "vanilla",
|
"prefix": "vanilla",
|
||||||
"order": this_settings.order,
|
"order": this_settings.order,
|
||||||
|
@ -1450,6 +1450,8 @@ app.post("/generate_novelai", jsonParser, async function (request, response_gene
|
|||||||
});
|
});
|
||||||
|
|
||||||
console.log(request.body);
|
console.log(request.body);
|
||||||
|
const bw = require('./src/bad-words');
|
||||||
|
const bad_words_ids = request.body.model.includes('clio') ? bw.clioBadWordsId : bw.badWordIds;
|
||||||
const data = {
|
const data = {
|
||||||
"input": request.body.input,
|
"input": request.body.input,
|
||||||
"model": request.body.model,
|
"model": request.body.model,
|
||||||
@ -1469,10 +1471,10 @@ app.post("/generate_novelai", jsonParser, async function (request, response_gene
|
|||||||
"top_k": request.body.top_k,
|
"top_k": request.body.top_k,
|
||||||
"typical_p": request.body.typical_p,
|
"typical_p": request.body.typical_p,
|
||||||
//"stop_sequences": {{187}},
|
//"stop_sequences": {{187}},
|
||||||
//bad_words_ids = {{50256}, {0}, {1}};
|
"bad_words_ids": bad_words_ids,
|
||||||
//generate_until_sentence = true;
|
//generate_until_sentence = true;
|
||||||
"use_cache": request.body.use_cache,
|
"use_cache": request.body.use_cache,
|
||||||
//use_string = true;
|
"use_string": true,
|
||||||
"return_full_text": request.body.return_full_text,
|
"return_full_text": request.body.return_full_text,
|
||||||
"prefix": request.body.prefix,
|
"prefix": request.body.prefix,
|
||||||
"order": request.body.order
|
"order": request.body.order
|
||||||
|
326
src/bad-words.js
Normal file
326
src/bad-words.js
Normal file
@ -0,0 +1,326 @@
|
|||||||
|
const badWordIds = [
|
||||||
|
[60],
|
||||||
|
[62],
|
||||||
|
[544],
|
||||||
|
[683],
|
||||||
|
[696],
|
||||||
|
[880],
|
||||||
|
[905],
|
||||||
|
[1008],
|
||||||
|
[1019],
|
||||||
|
[1084],
|
||||||
|
[1092],
|
||||||
|
[1181],
|
||||||
|
[1184],
|
||||||
|
[1254],
|
||||||
|
[1447],
|
||||||
|
[1570],
|
||||||
|
[1656],
|
||||||
|
[2194],
|
||||||
|
[2470],
|
||||||
|
[2479],
|
||||||
|
[2498],
|
||||||
|
[2947],
|
||||||
|
[3138],
|
||||||
|
[3291],
|
||||||
|
[3455],
|
||||||
|
[3725],
|
||||||
|
[3851],
|
||||||
|
[3891],
|
||||||
|
[3921],
|
||||||
|
[3951],
|
||||||
|
[4207],
|
||||||
|
[4299],
|
||||||
|
[4622],
|
||||||
|
[4681],
|
||||||
|
[5013],
|
||||||
|
[5032],
|
||||||
|
[5180],
|
||||||
|
[5218],
|
||||||
|
[5290],
|
||||||
|
[5413],
|
||||||
|
[5456],
|
||||||
|
[5709],
|
||||||
|
[5749],
|
||||||
|
[5774],
|
||||||
|
[6038],
|
||||||
|
[6257],
|
||||||
|
[6334],
|
||||||
|
[6660],
|
||||||
|
[6904],
|
||||||
|
[7082],
|
||||||
|
[7086],
|
||||||
|
[7254],
|
||||||
|
[7444],
|
||||||
|
[7748],
|
||||||
|
[8001],
|
||||||
|
[8088],
|
||||||
|
[8168],
|
||||||
|
[8562],
|
||||||
|
[8605],
|
||||||
|
[8795],
|
||||||
|
[8850],
|
||||||
|
[9014],
|
||||||
|
[9102],
|
||||||
|
[9259],
|
||||||
|
[9318],
|
||||||
|
[9336],
|
||||||
|
[9502],
|
||||||
|
[9686],
|
||||||
|
[9793],
|
||||||
|
[9855],
|
||||||
|
[9899],
|
||||||
|
[9955],
|
||||||
|
[10148],
|
||||||
|
[10174],
|
||||||
|
[10943],
|
||||||
|
[11326],
|
||||||
|
[11337],
|
||||||
|
[11661],
|
||||||
|
[12004],
|
||||||
|
[12084],
|
||||||
|
[12159],
|
||||||
|
[12520],
|
||||||
|
[12977],
|
||||||
|
[13380],
|
||||||
|
[13488],
|
||||||
|
[13663],
|
||||||
|
[13811],
|
||||||
|
[13976],
|
||||||
|
[14412],
|
||||||
|
[14598],
|
||||||
|
[14767],
|
||||||
|
[15640],
|
||||||
|
[15707],
|
||||||
|
[15775],
|
||||||
|
[15830],
|
||||||
|
[16079],
|
||||||
|
[16354],
|
||||||
|
[16369],
|
||||||
|
[16445],
|
||||||
|
[16595],
|
||||||
|
[16614],
|
||||||
|
[16731],
|
||||||
|
[16943],
|
||||||
|
[17278],
|
||||||
|
[17281],
|
||||||
|
[17548],
|
||||||
|
[17555],
|
||||||
|
[17981],
|
||||||
|
[18022],
|
||||||
|
[18095],
|
||||||
|
[18297],
|
||||||
|
[18413],
|
||||||
|
[18736],
|
||||||
|
[18772],
|
||||||
|
[18990],
|
||||||
|
[19181],
|
||||||
|
[20095],
|
||||||
|
[20197],
|
||||||
|
[20481],
|
||||||
|
[20629],
|
||||||
|
[20871],
|
||||||
|
[20879],
|
||||||
|
[20924],
|
||||||
|
[20977],
|
||||||
|
[21375],
|
||||||
|
[21382],
|
||||||
|
[21391],
|
||||||
|
[21687],
|
||||||
|
[21810],
|
||||||
|
[21828],
|
||||||
|
[21938],
|
||||||
|
[22367],
|
||||||
|
[22372],
|
||||||
|
[22734],
|
||||||
|
[23405],
|
||||||
|
[23505],
|
||||||
|
[23734],
|
||||||
|
[23741],
|
||||||
|
[23781],
|
||||||
|
[24237],
|
||||||
|
[24254],
|
||||||
|
[24345],
|
||||||
|
[24430],
|
||||||
|
[25416],
|
||||||
|
[25896],
|
||||||
|
[26119],
|
||||||
|
[26635],
|
||||||
|
[26842],
|
||||||
|
[26991],
|
||||||
|
[26997],
|
||||||
|
[27075],
|
||||||
|
[27114],
|
||||||
|
[27468],
|
||||||
|
[27501],
|
||||||
|
[27618],
|
||||||
|
[27655],
|
||||||
|
[27720],
|
||||||
|
[27829],
|
||||||
|
[28052],
|
||||||
|
[28118],
|
||||||
|
[28231],
|
||||||
|
[28532],
|
||||||
|
[28571],
|
||||||
|
[28591],
|
||||||
|
[28653],
|
||||||
|
[29013],
|
||||||
|
[29547],
|
||||||
|
[29650],
|
||||||
|
[29925],
|
||||||
|
[30522],
|
||||||
|
[30537],
|
||||||
|
[30996],
|
||||||
|
[31011],
|
||||||
|
[31053],
|
||||||
|
[31096],
|
||||||
|
[31148],
|
||||||
|
[31258],
|
||||||
|
[31350],
|
||||||
|
[31379],
|
||||||
|
[31422],
|
||||||
|
[31789],
|
||||||
|
[31830],
|
||||||
|
[32214],
|
||||||
|
[32666],
|
||||||
|
[32871],
|
||||||
|
[33094],
|
||||||
|
[33376],
|
||||||
|
[33440],
|
||||||
|
[33805],
|
||||||
|
[34368],
|
||||||
|
[34398],
|
||||||
|
[34417],
|
||||||
|
[34418],
|
||||||
|
[34419],
|
||||||
|
[34476],
|
||||||
|
[34494],
|
||||||
|
[34607],
|
||||||
|
[34758],
|
||||||
|
[34761],
|
||||||
|
[34904],
|
||||||
|
[34993],
|
||||||
|
[35117],
|
||||||
|
[35138],
|
||||||
|
[35237],
|
||||||
|
[35487],
|
||||||
|
[35830],
|
||||||
|
[35869],
|
||||||
|
[36033],
|
||||||
|
[36134],
|
||||||
|
[36320],
|
||||||
|
[36399],
|
||||||
|
[36487],
|
||||||
|
[36586],
|
||||||
|
[36676],
|
||||||
|
[36692],
|
||||||
|
[36786],
|
||||||
|
[37077],
|
||||||
|
[37594],
|
||||||
|
[37596],
|
||||||
|
[37786],
|
||||||
|
[37982],
|
||||||
|
[38475],
|
||||||
|
[38791],
|
||||||
|
[39083],
|
||||||
|
[39258],
|
||||||
|
[39487],
|
||||||
|
[39822],
|
||||||
|
[40116],
|
||||||
|
[40125],
|
||||||
|
[41000],
|
||||||
|
[41018],
|
||||||
|
[41256],
|
||||||
|
[41305],
|
||||||
|
[41361],
|
||||||
|
[41447],
|
||||||
|
[41449],
|
||||||
|
[41512],
|
||||||
|
[41604],
|
||||||
|
[42041],
|
||||||
|
[42274],
|
||||||
|
[42368],
|
||||||
|
[42696],
|
||||||
|
[42767],
|
||||||
|
[42804],
|
||||||
|
[42854],
|
||||||
|
[42944],
|
||||||
|
[42989],
|
||||||
|
[43134],
|
||||||
|
[43144],
|
||||||
|
[43189],
|
||||||
|
[43521],
|
||||||
|
[43782],
|
||||||
|
[44082],
|
||||||
|
[44162],
|
||||||
|
[44270],
|
||||||
|
[44308],
|
||||||
|
[44479],
|
||||||
|
[44524],
|
||||||
|
[44965],
|
||||||
|
[45114],
|
||||||
|
[45301],
|
||||||
|
[45382],
|
||||||
|
[45443],
|
||||||
|
[45472],
|
||||||
|
[45488],
|
||||||
|
[45507],
|
||||||
|
[45564],
|
||||||
|
[45662],
|
||||||
|
[46265],
|
||||||
|
[46267],
|
||||||
|
[46275],
|
||||||
|
[46295],
|
||||||
|
[46462],
|
||||||
|
[46468],
|
||||||
|
[46576],
|
||||||
|
[46694],
|
||||||
|
[47093],
|
||||||
|
[47384],
|
||||||
|
[47389],
|
||||||
|
[47446],
|
||||||
|
[47552],
|
||||||
|
[47686],
|
||||||
|
[47744],
|
||||||
|
[47916],
|
||||||
|
[48064],
|
||||||
|
[48167],
|
||||||
|
[48392],
|
||||||
|
[48471],
|
||||||
|
[48664],
|
||||||
|
[48701],
|
||||||
|
[49021],
|
||||||
|
[49193],
|
||||||
|
[49236],
|
||||||
|
[49550],
|
||||||
|
[49694],
|
||||||
|
[49806],
|
||||||
|
[49824],
|
||||||
|
[50001],
|
||||||
|
[50256],
|
||||||
|
[0],
|
||||||
|
[1],
|
||||||
|
]
|
||||||
|
|
||||||
|
const clioBadWordsId = [
|
||||||
|
[3],
|
||||||
|
[49356],
|
||||||
|
[1431],
|
||||||
|
[31715],
|
||||||
|
[34387],
|
||||||
|
[20765],
|
||||||
|
[30702],
|
||||||
|
[10691],
|
||||||
|
[49333],
|
||||||
|
[1266],
|
||||||
|
[19438],
|
||||||
|
[43145],
|
||||||
|
[26523],
|
||||||
|
[41471],
|
||||||
|
[2936],
|
||||||
|
]
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
badWordIds,
|
||||||
|
clioBadWordsId,
|
||||||
|
};
|
Reference in New Issue
Block a user