fix sdxl brushnet load sdxl single file

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

View File

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

View File

@@ -173,6 +173,10 @@ class ModelManager:
}
if hasattr(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.name,