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