From a84f7c4a58edf90a767bc5cb40bc070ef4aff8c7 Mon Sep 17 00:00:00 2001 From: yolain Date: Sat, 11 May 2024 00:42:14 +0800 Subject: [PATCH] fix:brushnet error in easy presamplingInpainting --- py/easyNodes.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/py/easyNodes.py b/py/easyNodes.py index 4327f24..4264d6e 100644 --- a/py/easyNodes.py +++ b/py/easyNodes.py @@ -4774,11 +4774,11 @@ class samplerSimpleInpainting: brushname = os.path.basename(parsed_url.path) return brushname - def apply_brushnet(self, brushname, model, vae, image, mask, positive, negative, scale=1.0, start_at=0.0, end_at=1.0): + def apply_brushnet(self, brushname, model, vae, image, mask, positive, negative, scale=1.0, start_at=0, end_at=10000): if "BrushNetLoader" not in ALL_NODE_CLASS_MAPPINGS: raise Exception("BrushNetLoader not found,please install ComfyUI-BrushNet") cls = ALL_NODE_CLASS_MAPPINGS['BrushNetLoader'] - brushnet, = cls().brushnet_loading(brushname) + brushnet, = cls().brushnet_loading(brushname, 'float16') cls = ALL_NODE_CLASS_MAPPINGS['BrushNet'] m, positive, negative, latent = cls().model_update(model=model, vae=vae, image=image, mask=mask, brushnet=brushnet, positive=positive, negative=negative, scale=scale, start_at=start_at, end_at=end_at) return m, positive, negative, latent