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],