change base model

app.py

@@ -21,7 +21,8 @@ from config import (
     MINIMUM_IMAGE_NUMBER,
     MAXIMUM_IMAGE_NUMBER,
     DEFAULT_NEGATIVE_PROMPT,
-    DEFAULT_POSITIVE_PROMPT
+    DEFAULT_POSITIVE_PROMPT,
+    task_stablepy
 )
 from models.vae import VAE_LIST as download_vae
 from models.checkpoints import CHECKPOINT_LIST as download_model
@@ -34,30 +35,6 @@ from examples.examples import example_prompts
 from utils.download_utils import download_things
 from utils.model_utils import get_model_list
 
-task_stablepy: dict = {
-    'txt2img': 'txt2img',
-    'img2img': 'img2img',
-    'inpaint': 'inpaint',
-    # 'canny T2I Adapter': 'sdxl_canny_t2i',  # NO HAVE STEP CALLBACK PARAMETERS SO NOT WORKS WITH DIFFUSERS 0.29.0
-    # 'sketch  T2I Adapter': 'sdxl_sketch_t2i',
-    # 'lineart  T2I Adapter': 'sdxl_lineart_t2i',
-    # 'depth-midas  T2I Adapter': 'sdxl_depth-midas_t2i',
-    # 'openpose  T2I Adapter': 'sdxl_openpose_t2i',
-    'openpose ControlNet': 'openpose',
-    'canny ControlNet': 'canny',
-    'mlsd ControlNet': 'mlsd',
-    'scribble ControlNet': 'scribble',
-    'softedge ControlNet': 'softedge',
-    'segmentation ControlNet': 'segmentation',
-    'depth ControlNet': 'depth',
-    'normalbae ControlNet': 'normalbae',
-    'lineart ControlNet': 'lineart',
-    # 'lineart_anime ControlNet': 'lineart_anime',
-    'shuffle ControlNet': 'shuffle',
-    'ip2p ControlNet': 'ip2p',
-    'optical pattern ControlNet': 'pattern',
-    'tile realistic': 'sdxl_tile_realistic',
-}
 # LOAD ALL ENV TOKEN
 CIVITAI_API_KEY: str = os.environ.get("CIVITAI_API_KEY")
 hf_token: str = os.environ.get("HF_TOKEN")

config.py

@@ -28,3 +28,28 @@ cross eyed, cloned face, bad symmetry, bad anatomy, low quality, blurry, text, w
 signature, jpeg, artifacts, monochrome, paintings, oil, (hands:1.15), European Woman, woman, 
 noise, dark skin, (3d), By bad artist -neg,bhands-neg, canvas frame, (badly drawn), (badly painted),
 """
+
+task_stablepy: dict = {
+    'txt2img': 'txt2img',
+    'img2img': 'img2img',
+    'inpaint': 'inpaint',
+    # 'canny T2I Adapter': 'sdxl_canny_t2i',  # NO HAVE STEP CALLBACK PARAMETERS SO NOT WORKS WITH DIFFUSERS 0.29.0
+    # 'sketch  T2I Adapter': 'sdxl_sketch_t2i',
+    # 'lineart  T2I Adapter': 'sdxl_lineart_t2i',
+    # 'depth-midas  T2I Adapter': 'sdxl_depth-midas_t2i',
+    # 'openpose  T2I Adapter': 'sdxl_openpose_t2i',
+    'openpose ControlNet': 'openpose',
+    'canny ControlNet': 'canny',
+    'mlsd ControlNet': 'mlsd',
+    'scribble ControlNet': 'scribble',
+    'softedge ControlNet': 'softedge',
+    'segmentation ControlNet': 'segmentation',
+    'depth ControlNet': 'depth',
+    'normalbae ControlNet': 'normalbae',
+    'lineart ControlNet': 'lineart',
+    # 'lineart_anime ControlNet': 'lineart_anime',
+    'shuffle ControlNet': 'shuffle',
+    'ip2p ControlNet': 'ip2p',
+    'optical pattern ControlNet': 'pattern',
+    'tile realistic': 'sdxl_tile_realistic',
+}

gui.py

@@ -213,7 +213,7 @@ class GuiSD:
                 print(la)
                 lora_type = ("animetarot" in la.lower() or "Hyper-SD15-8steps".lower() in la.lower())
                 if (model_is_xl and lora_type) or (not model_is_xl and not lora_type):
-                    msg_inc_lora = f'The LoRA {la} is for {'SD 1.5' if model_is_xl else 'SDXL'}, but you are using {model_type}.'
+                    msg_inc_lora = f"The LoRA {la} is for {'SD 1.5' if model_is_xl else 'SDXL'}, but you are using {model_type}."
                     gr.Info(msg_inc_lora)
                     msg_lora.append(msg_inc_lora)