nodes/smthemex-ComfyUI_MS_Diffusion
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update MS_Diffusion.py

Browse Source
This commit is contained in:
smthemex
2024-09-06 15:34:48 +08:00
committed by GitHub
parent a1698eaadd
commit e958a0fd27
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
MS_Diffusion.py
+5 -4
View File
@@ -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"