From e753246373847da681ccec001cddd090a933e70f Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Thu, 16 Nov 2023 02:15:28 +0200
Subject: [PATCH] Add NAI Diffusion 3 model

---
 public/scripts/extensions/stable-diffusion/index.js |  4 ++++
 src/novelai.js                                      | 10 +++++++++-
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/public/scripts/extensions/stable-diffusion/index.js b/public/scripts/extensions/stable-diffusion/index.js
index c859fc250..5984a603d 100644
--- a/public/scripts/extensions/stable-diffusion/index.js
+++ b/public/scripts/extensions/stable-diffusion/index.js
@@ -1180,6 +1180,10 @@ async function loadNovelModels() {
     }
 
     return [
+        {
+            value: 'nai-diffusion-3',
+            text: 'NAI Diffusion Anime V3',
+        },
         {
             value: 'nai-diffusion-2',
             text: 'NAI Diffusion Anime V2',
diff --git a/src/novelai.js b/src/novelai.js
index e4d9bffe3..192954391 100644
--- a/src/novelai.js
+++ b/src/novelai.js
@@ -268,12 +268,20 @@ function registerEndpoints(app, jsonParser) {
                         // NAI handholding for prompts
                         ucPreset: 0,
                         qualityToggle: false,
+                        add_original_image: false,
+                        controlnet_strength: 1,
+                        dynamic_thresholding: false,
+                        legacy: false,
+                        sm: false,
+                        sm_dyn: false,
+                        uncond_scale: 1,
                     },
                 }),
             });
 
             if (!generateResult.ok) {
-                console.log('NovelAI returned an error.', generateResult.statusText);
+                const text = await generateResult.text();
+                console.log('NovelAI returned an error.', generateResult.statusText, text);
                 return response.sendStatus(500);
             }