fix sdxl brushnet load sdxl single file

2025-03-17 15:48:58 +08:00
parent 212dd3ed69
commit 73a4bf4456
2 changed files with 4 additions and 2 deletions
--- 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,