NovelAI: Fix scheduler auto-assign

d0e1c3287a828eff65a2c92380aaad9c38531cee

Cohee <18619528+Cohee1207@users.noreply.github.com>

1 files changed, +7 -2Ignore whitespace
public/scripts/extensions/stable-diffusion/index.js+7 -2
@@ -1994,6 +1994,10 @@ async function loadNovelModels() {
19941994 ];
19951995}
19961996
1997+function loadNovelSchedulers() {
1998+ return ['karras', 'native', 'exponential', 'polyexponential'];
1999+}
2000+
19972001async function loadComfyModels() {
19982002 if (!extension_settings.sd.comfy_url) {
19992003 return [];
@@ -2031,7 +2035,7 @@ async function loadSchedulers() {
20312035 schedulers = await getAutoRemoteSchedulers();
20322036 break;
20332037 case sources.novel:
2034- schedulers = ['karras', 'native', 'exponential', 'polyexponential'];
2038+ schedulers = loadNovelSchedulers();
20352039 break;
20362040 case sources.vlad:
20372041 schedulers = ['N/A'];
@@ -3177,7 +3181,8 @@ function getNovelParams() {
31773181 let sm_dyn = extension_settings.sd.novel_sm_dyn;
31783182
31793183 // If a source was never changed after the scheduler setting was added, we need to set it to 'karras' for compatibility.
3180- if (!extension_settings.sd.scheduler || extension_settings.sd.scheduler === 'normal') {
3184+ const schedulers = loadNovelSchedulers();
3185+ if (!schedulers.includes(extension_settings.sd.scheduler)) {
31813186 extension_settings.sd.scheduler = 'karras';
31823187 }
31833188