fix sdxl brushnet load sdxl single file
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user