From acfdd7713c7fb4793af0accba76741bf2b72c7b5 Mon Sep 17 00:00:00 2001 From: yolain Date: Thu, 21 Nov 2024 17:09:00 +0800 Subject: [PATCH] Fix brushnet can not be used with startup arg `--fast` mode --- README.ZH_CN.md | 1 + README.md | 1 + py/brushnet/__init__.py | 4 ++-- 3 files changed, 4 insertions(+), 2 deletions(-) diff --git a/README.ZH_CN.md b/README.ZH_CN.md index fd35b3c..b77431f 100644 --- a/README.ZH_CN.md +++ b/README.ZH_CN.md @@ -63,6 +63,7 @@ git clone https://github.com/yolain/ComfyUI-Easy-Use **v1.2.5** +- 修复brushnet 无法在 `--fast` 模式下使用 - 支持briaai RMBG-2.0 - 支持mochi模型 - 实现在循环主体中重复使用终端节点输出（例如预览图像和显示任何内容等输出节点...） diff --git a/README.md b/README.md index ca51e54..3659eee 100644 --- a/README.md +++ b/README.md @@ -58,6 +58,7 @@ Double-click install.bat to install the required dependencies **v1.2.5** +- Fix brushnet can not be used with startup arg `--fast` mode - Support briaai RMBG-2.0 - Support mochi - Implement reuse of end nodes output in the loop body (e.g: previewImage and showAnything and sth.) diff --git a/py/brushnet/__init__.py b/py/brushnet/__init__.py index 5955fe9..b700042 100644 --- a/py/brushnet/__init__.py +++ b/py/brushnet/__init__.py @@ -664,7 +664,7 @@ def add_brushnet_patch(model, brushnet, torch_dtype, conditioning_latents, is_SDXL = isinstance(model.model.model_config, comfy.supported_models.SDXL) if is_SDXL: - input_blocks = [[0, comfy.ops.disable_weight_init.Conv2d], + input_blocks = [[0, comfy.ops.manual_cast.Conv2d], [1, comfy.ldm.modules.diffusionmodules.openaimodel.ResBlock], [2, comfy.ldm.modules.diffusionmodules.openaimodel.ResBlock], [3, comfy.ldm.modules.diffusionmodules.openaimodel.Downsample], @@ -686,7 +686,7 @@ def add_brushnet_patch(model, brushnet, torch_dtype, conditioning_latents, [7, comfy.ldm.modules.diffusionmodules.openaimodel.ResBlock], [8, comfy.ldm.modules.diffusionmodules.openaimodel.ResBlock]] else: - input_blocks = [[0, comfy.ops.disable_weight_init.Conv2d], + input_blocks = [[0, comfy.ops.manual_cast.Conv2d], [1, comfy.ldm.modules.attention.SpatialTransformer], [2, comfy.ldm.modules.attention.SpatialTransformer], [3, comfy.ldm.modules.diffusionmodules.openaimodel.Downsample],