let5see/IOPaint
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix sdxl brushnet load sdxl single file

Browse Source
This commit is contained in:
Qing
2025-03-17 15:48:58 +08:00
parent 212dd3ed69
commit 73a4bf4456
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
iopaint/model/brushnet/brushnet_xl_wrapper.py
View File

@@ -26,10 +26,8 @@ from ...const import SDXL_BRUSHNET_CHOICES
class BrushNetXLWrapper(DiffusionInpaintModel): class BrushNetXLWrapper(DiffusionInpaintModel):
name = "RunDiffusion/Juggernaut-XI-v11"
pad_mod = 8 pad_mod = 8
min_size = 1024 min_size = 1024
model_id_or_path = "RunDiffusion/Juggernaut-XI-v11"
support_brushnet = True support_brushnet = True
support_lcm_lora = False support_lcm_lora = False

4
iopaint/model_manager.py
View File

@@ -173,6 +173,10 @@ class ModelManager:
} }
if hasattr(self.model.model, "text_encoder_2"): if hasattr(self.model.model, "text_encoder_2"):
pipe_components["text_encoder_2"] = self.model.model.text_encoder_2 pipe_components["text_encoder_2"] = self.model.model.text_encoder_2
if hasattr(self.model.model, "tokenizer"):
pipe_components["tokenizer"] = self.model.model.tokenizer
if hasattr(self.model.model, "tokenizer_2"):
pipe_components["tokenizer_2"] = self.model.model.tokenizer_2
self.model = self.init_model( self.model = self.init_model(
self.name, self.name,