From 3f026aea15b6181fac5edb1bc12ee39424981db2 Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Fri, 23 Aug 2024 19:57:25 +0300
Subject: [PATCH] SD: Add prompt expansion warning. Disable quantization

---
 public/scripts/extensions/stable-diffusion/settings.html | 3 ++-
 src/transformers.mjs                                     | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/public/scripts/extensions/stable-diffusion/settings.html b/public/scripts/extensions/stable-diffusion/settings.html
index b89366b7c..821b5f791 100644
--- a/public/scripts/extensions/stable-diffusion/settings.html
+++ b/public/scripts/extensions/stable-diffusion/settings.html
@@ -29,7 +29,8 @@
             </label>
             <label for="sd_expand" class="checkbox_label" data-i18n="[title]sd_expand" title="Automatically extend prompts using text generation model">
                 <input id="sd_expand" type="checkbox" />
-                <span data-i18n="sd_expand_txt">Auto-enhance prompts</span>
+                <span data-i18n="sd_expand_txt">Auto-extend prompts</span>
+                <span class="right_menu_button fa-solid fa-triangle-exclamation" data-i18n="[title]sd_expand_warning" title="May produce unexpected results. Manual prompt editing is recommended."></span>
             </label>
             <label for="sd_snap" class="checkbox_label" data-i18n="[title]sd_snap" title="Snap generation requests with a forced aspect ratio (portraits, backgrounds) to the nearest known resolution, while trying to preserve the absolute pixel counts (recommended for SDXL).">
                 <input id="sd_snap" type="checkbox" />
diff --git a/src/transformers.mjs b/src/transformers.mjs
index f3c7389a4..fe947123a 100644
--- a/src/transformers.mjs
+++ b/src/transformers.mjs
@@ -35,7 +35,7 @@ const tasks = {
         defaultModel: 'Cohee/fooocus_expansion-onnx',
         pipeline: null,
         configField: 'extras.promptExpansionModel',
-        quantized: true,
+        quantized: false,
     },
     'automatic-speech-recognition': {
         defaultModel: 'Xenova/whisper-small',