rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

NovelAI: Fix scheduler auto-assign

Browse Source
This commit is contained in:
Cohee
2025-01-02 22:58:06 +02:00
parent beb4328714
commit d0e1c3287a
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
public/scripts/extensions/stable-diffusion/index.js
View File

@ -1994,6 +1994,10 @@ async function loadNovelModels() {
]; ];
} }
function loadNovelSchedulers() {
return ['karras', 'native', 'exponential', 'polyexponential'];
}
async function loadComfyModels() { async function loadComfyModels() {
if (!extension_settings.sd.comfy_url) { if (!extension_settings.sd.comfy_url) {
return []; return [];
@ -2031,7 +2035,7 @@ async function loadSchedulers() {
schedulers = await getAutoRemoteSchedulers(); schedulers = await getAutoRemoteSchedulers();
break; break;
case sources.novel: case sources.novel:
schedulers = ['karras', 'native', 'exponential', 'polyexponential']; schedulers = loadNovelSchedulers();
break; break;
case sources.vlad: case sources.vlad:
schedulers = ['N/A']; schedulers = ['N/A'];
@ -3177,7 +3181,8 @@ function getNovelParams() {
let sm_dyn = extension_settings.sd.novel_sm_dyn; let sm_dyn = extension_settings.sd.novel_sm_dyn;
// If a source was never changed after the scheduler setting was added, we need to set it to 'karras' for compatibility. // If a source was never changed after the scheduler setting was added, we need to set it to 'karras' for compatibility.
if (!extension_settings.sd.scheduler || extension_settings.sd.scheduler === 'normal') { const schedulers = loadNovelSchedulers();
if (!schedulers.includes(extension_settings.sd.scheduler)) {
extension_settings.sd.scheduler = 'karras'; extension_settings.sd.scheduler = 'karras';
} }