From c42a7739d6d1c85e8a5fdc276185194eea2585c3 Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Fri, 27 Dec 2024 00:24:53 +0200
Subject: [PATCH] NovelAI: Fix backward compatibility with scheduler setting

---
 public/scripts/extensions/stable-diffusion/index.js | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/public/scripts/extensions/stable-diffusion/index.js b/public/scripts/extensions/stable-diffusion/index.js
index 0ac7ffa24..f8bca0e73 100644
--- a/public/scripts/extensions/stable-diffusion/index.js
+++ b/public/scripts/extensions/stable-diffusion/index.js
@@ -3174,6 +3174,11 @@ function getNovelParams() {
     let sm = extension_settings.sd.novel_sm;
     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 (!extension_settings.sd.scheduler || extension_settings.sd.scheduler === 'normal') {
+        extension_settings.sd.scheduler = 'karras';
+    }
+
     if (extension_settings.sd.sampler === 'ddim' || extension_settings.sd.model === 'nai-diffusion-4-curated-preview') {
         sm = false;
         sm_dyn = false;