From e958a0fd279f0ae6fe236b235cef668eccd2543f Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Fri, 6 Sep 2024 15:34:48 +0800
Subject: [PATCH] Update MS_Diffusion.py

---
 MS_Diffusion.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/MS_Diffusion.py b/MS_Diffusion.py
index 79ecacf..c22ceef 100644
--- a/MS_Diffusion.py
+++ b/MS_Diffusion.py
@@ -289,6 +289,7 @@ class MSdiffusion_Model_Loader:
             lora = lora.split("\\")[-1]
         
         # load SDXL pipeline
+        add_config="stabilityai/stable-diffusion-xl-base-1.0"
         if not repo_id:
             original_config_file = os.path.join(dir_path, "config", "sd_xl_base.yaml")
             
@@ -298,12 +299,12 @@ class MSdiffusion_Model_Loader:
                     torch_dtype=torch.float16
                 )
                 try:
-                    pipe = StableDiffusionXLControlNetPipeline.from_single_file(ckpt_path, controlnet=controlnet,
+                    pipe = StableDiffusionXLControlNetPipeline.from_single_file(ckpt_path,config=add_config, controlnet=controlnet,
                                                                                 original_config=original_config_file,
                                                                                 torch_dtype=torch.float16, )
                 except:
                     try:
-                        pipe = StableDiffusionXLControlNetPipeline.from_single_file(ckpt_path, controlnet=controlnet,
+                        pipe = StableDiffusionXLControlNetPipeline.from_single_file(ckpt_path,config=add_config, controlnet=controlnet,
                                                                                     original_config_file=original_config_file,
                                                                                     torch_dtype=torch.float16, )
                     except:
@@ -311,11 +312,11 @@ class MSdiffusion_Model_Loader:
             else:
                 try:
                     pipe = StableDiffusionXLPipeline.from_single_file(
-                        ckpt_path, original_config=original_config_file, torch_dtype=torch.float16)
+                        ckpt_path,config=add_config, original_config=original_config_file, torch_dtype=torch.float16)
                 except:
                     try:
                         pipe = StableDiffusionXLPipeline.from_single_file(
-                            ckpt_path, original_config_file=original_config_file, torch_dtype=torch.float16)
+                            ckpt_path,config=add_config, original_config_file=original_config_file, torch_dtype=torch.float16)
                     except:
                         raise "model error"