298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-11 02:15:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update nodes_differential_diffusion.py

Browse Source
This commit is contained in:
Eric 2024-11-21 23:43:17 -05:00 committed by GitHub
parent ce31337813
commit 1a864435f6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy_extras/nodes_differential_diffusion.py
View File

@ -43,9 +43,11 @@ class DifferentialDiffusion():
binary_mask = (denoise_mask >= threshold).to(denoise_mask.dtype) binary_mask = (denoise_mask >= threshold).to(denoise_mask.dtype)
# Blend binary mask with the original denoise_mask using strength # Blend binary mask with the original denoise_mask using strength
blended_mask = strength * binary_mask + (1 - strength) * denoise_mask if strength and strength < 1:
blended_mask = strength * binary_mask + (1 - strength) * denoise_mask
return blended_mask return blended_mask
else:
return binary_mask
NODE_CLASS_MAPPINGS = { NODE_CLASS_MAPPINGS = {