From f9ae7ea949ede12aa840a1ec4ce3ab33123df595 Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Mon, 23 Oct 2023 01:06:49 +0300
Subject: [PATCH] Add NAI Diffusion v2 model

---
 public/scripts/extensions/stable-diffusion/index.js | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/public/scripts/extensions/stable-diffusion/index.js b/public/scripts/extensions/stable-diffusion/index.js
index 21289ef0e..4468ae912 100644
--- a/public/scripts/extensions/stable-diffusion/index.js
+++ b/public/scripts/extensions/stable-diffusion/index.js
@@ -1081,17 +1081,21 @@ async function loadNovelModels() {
     }
 
     return [
+        {
+            value: 'nai-diffusion-2',
+            text: 'NAI Diffusion Anime V2',
+        },
         {
             value: 'nai-diffusion',
-            text: 'Full',
+            text: 'NAI Diffusion Anime V1 (Full)',
         },
         {
             value: 'safe-diffusion',
-            text: 'Safe',
+            text: 'NAI Diffusion Anime V1 (Curated)',
         },
         {
             value: 'nai-diffusion-furry',
-            text: 'Furry',
+            text: 'NAI Diffusion Furry',
         },
     ];
 }