From 73a4bf44569bee1e151816b085359753b783a11f Mon Sep 17 00:00:00 2001 From: Qing Date: Mon, 17 Mar 2025 15:48:58 +0800 Subject: [PATCH] fix sdxl brushnet load sdxl single file --- iopaint/model/brushnet/brushnet_xl_wrapper.py | 2 -- iopaint/model_manager.py | 4 ++++ 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/iopaint/model/brushnet/brushnet_xl_wrapper.py b/iopaint/model/brushnet/brushnet_xl_wrapper.py index ed35a2e..721e8ce 100644 --- a/iopaint/model/brushnet/brushnet_xl_wrapper.py +++ b/iopaint/model/brushnet/brushnet_xl_wrapper.py @@ -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 diff --git a/iopaint/model_manager.py b/iopaint/model_manager.py index 90d9c83..2abc60e 100644 --- a/iopaint/model_manager.py +++ b/iopaint/model_manager.py @@ -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,