From a899a02d301eb5296f37545c541a48e4f9a03c39 Mon Sep 17 00:00:00 2001
From: niceandneat <niceaneat@gmail.com>
Date: Mon, 28 Apr 2025 01:19:47 +0900
Subject: [PATCH] Add nai cfg_rescale, noise_schedule options settings

---
 src/lib/Setting/Pages/OtherBotSettings.svelte | 13 +++++++++++++
 src/ts/process/stableDiff.ts                  |  4 ++--
 src/ts/storage/database.svelte.ts             |  4 ++++
 3 files changed, 19 insertions(+), 2 deletions(-)

diff --git a/src/lib/Setting/Pages/OtherBotSettings.svelte b/src/lib/Setting/Pages/OtherBotSettings.svelte
index 13b8e9bd..474eb59b 100644
--- a/src/lib/Setting/Pages/OtherBotSettings.svelte
+++ b/src/lib/Setting/Pages/OtherBotSettings.svelte
@@ -26,8 +26,10 @@
             width: 512,
             height: 512,
             sampler: 'k_euler',
+            noise_schedule: 'native',
             steps: 100,
             scale: 1,
+            cfg_rescale: 0,
             sm: false,
             sm_dyn: false,
             strength: 0.5,
@@ -245,10 +247,21 @@
                 </SelectInput>
             {/if}
 
+            <span class="text-textcolor">Noise Schedule</span>
+            <SelectInput className="mt-2 mb-4" bind:value={DBState.db.NAIImgConfig.noise_schedule}>
+                <OptionInput value="" >Choose...</OptionInput>
+                <OptionInput value="native" >native</OptionInput>
+                <OptionInput value="karras" >karras</OptionInput>
+                <OptionInput value="exponential" >exponential</OptionInput>
+                <OptionInput value="polyexponential" >polyexponential</OptionInput>
+            </SelectInput>
+
             <span class="text-textcolor">steps</span>
             <NumberInput size="sm" marginBottom min={0} max={2048} bind:value={DBState.db.NAIImgConfig.steps}/>
             <span class="text-textcolor">CFG scale</span>
             <NumberInput size="sm" marginBottom min={0} max={2048} bind:value={DBState.db.NAIImgConfig.scale}/>
+            <span class="text-textcolor">CFG rescale</span>
+            <NumberInput size="sm" marginBottom min={0} max={1} bind:value={DBState.db.NAIImgConfig.cfg_rescale}/>
 
             {#if !DBState.db.NAII2I || DBState.db.NAIImgConfig.sampler !== 'ddim_v3'}
                 <Check bind:check={DBState.db.NAIImgConfig.sm} name="Use SMEA"/>
diff --git a/src/ts/process/stableDiff.ts b/src/ts/process/stableDiff.ts
index fb24033a..34ed094e 100644
--- a/src/ts/process/stableDiff.ts
+++ b/src/ts/process/stableDiff.ts
@@ -132,7 +132,7 @@ export async function generateAIImage(genPrompt:string, currentChar:character, n
                 "parameters": {
                     "params_version": 3,
                     "add_original_image": true,
-                    "cfg_rescale": 0,
+                    "cfg_rescale": db.NAIImgConfig.cfg_rescale,
                     "controlnet_strength": 1,
                     "dynamic_thresholding": false,
                     "n_samples": 1,
@@ -145,7 +145,7 @@ export async function generateAIImage(genPrompt:string, currentChar:character, n
                     "sm": false,
                     "sm_dyn": false,
                     "noise": db.NAIImgConfig.noise,
-                    "noise_schedule": "native",
+                    "noise_schedule": db.NAIImgConfig.noise_schedule,
                     "strength": db.NAIImgConfig.strength,
                     "ucPreset": 3,
                     "uncond_scale": 1,
diff --git a/src/ts/storage/database.svelte.ts b/src/ts/storage/database.svelte.ts
index 1c0f6aad..49d9a5a8 100644
--- a/src/ts/storage/database.svelte.ts
+++ b/src/ts/storage/database.svelte.ts
@@ -255,8 +255,10 @@ export function setDatabase(data:Database){
             width:512,
             height:768,
             sampler:"k_dpmpp_sde",
+            noise_schedule:"native",
             steps:28,
             scale:5,
+            cfg_rescale: 0,
             sm:true,
             sm_dyn:false,
             noise:0.0,
@@ -1408,8 +1410,10 @@ export interface NAIImgConfig{
     width:number,
     height:number,
     sampler:string,
+    noise_schedule:string,
     steps:number,
     scale:number,
+    cfg_rescale:number,
     sm:boolean,
     sm_dyn:boolean,
     noise:number,