Fix default NAI presets. Add NAI bad words

This commit is contained in:
Cohee
2023-06-11 13:17:55 +03:00
parent c34455ef1e
commit 525b9dc0bb
5 changed files with 356 additions and 20 deletions

View File

@ -1450,6 +1450,8 @@ app.post("/generate_novelai", jsonParser, async function (request, response_gene
});
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 = {
"input": request.body.input,
"model": request.body.model,
@ -1469,10 +1471,10 @@ app.post("/generate_novelai", jsonParser, async function (request, response_gene
"top_k": request.body.top_k,
"typical_p": request.body.typical_p,
//"stop_sequences": {{187}},
//bad_words_ids = {{50256}, {0}, {1}};
"bad_words_ids": bad_words_ids,
//generate_until_sentence = true;
"use_cache": request.body.use_cache,
//use_string = true;
"use_string": true,
"return_full_text": request.body.return_full_text,
"prefix": request.body.prefix,
"order": request.body.order