From 368f7e508dec4f5b5ecb7492e9e2ea6e505a8097 Mon Sep 17 00:00:00 2001 From: yolain Date: Mon, 21 Apr 2025 11:00:38 +0800 Subject: [PATCH] EasyUse still works when brushnet-related diffusers error --- py/modules/brushnet/__init__.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/py/modules/brushnet/__init__.py b/py/modules/brushnet/__init__.py index e696684..c35ea13 100644 --- a/py/modules/brushnet/__init__.py +++ b/py/modules/brushnet/__init__.py @@ -5,13 +5,21 @@ import os import types import torch -from accelerate import init_empty_weights, load_checkpoint_and_dispatch +try: + from accelerate import init_empty_weights, load_checkpoint_and_dispatch +except: + init_empty_weights, load_checkpoint_and_dispatch = None, None import comfy -from .model import BrushNetModel, PowerPaintModel -from .model_patch import add_model_patch_option, patch_model_function_wrapper -from .powerpaint_utils import TokenizerWrapper, add_tokens +try: + from .model import BrushNetModel, PowerPaintModel + from .model_patch import add_model_patch_option, patch_model_function_wrapper + from .powerpaint_utils import TokenizerWrapper, add_tokens +except: + BrushNetModel, PowerPaintModel = None, None + add_model_patch_option, patch_model_function_wrapper = None, None + TokenizerWrapper, add_tokens = None, None cwd_path = os.path.dirname(os.path.realpath(__file__)) brushnet_config_file = os.path.join(cwd_path, 'config', 'brushnet.json')