rixty
/
SillyTavern
mirror of https://github.com/SillyTavern/SillyTavern.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix banned tokens not working for Erato

This commit is contained in:
Cohee 2024-09-24 07:55:23 +00:00
parent 182756fcb2
commit 8775247942
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/endpoints/novelai.js
View File

@ -116,10 +116,10 @@ router.post('/generate', jsonParser, async function (req, res) {
controller.abort(); controller.abort();
}); });
const isNewModel = (req.body.model.includes('clio') || req.body.model.includes('kayra')); const isNewModel = (req.body.model.includes('clio') || req.body.model.includes('kayra') || req.body.model.includes('erato'));
const badWordsList = getBadWordsList(req.body.model); const badWordsList = getBadWordsList(req.body.model);
// Add customized bad words for Clio and Kayra // Add customized bad words for Clio, Kayra, and Erato
if (isNewModel && Array.isArray(req.body.bad_words_ids)) { if (isNewModel && Array.isArray(req.body.bad_words_ids)) {
for (const badWord of req.body.bad_words_ids) { for (const badWord of req.body.bad_words_ids) {
if (Array.isArray(badWord) && badWord.every(x => Number.isInteger(x))) { if (Array.isArray(badWord) && badWord.every(x => Number.isInteger(x))) {
@ -183,7 +183,7 @@ router.post('/generate', jsonParser, async function (req, res) {
}; };
// Tells the model to stop generation at '>' // Tells the model to stop generation at '>'
if ('theme_textadventure' === req.body.prefix && isNewModel) { if ('theme_textadventure' === req.body.prefix && isNewModel && !req.body.model.includes('erato')) {
data.parameters.eos_token_id = 49405; data.parameters.eos_token_id = 49405;
} }