xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-21 02:34:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix outdated scipy erf import

Browse Source
This commit is contained in:
Kijai 2024-05-29 15:04:08 +03:00
parent d8cf496704
commit 0690bff42c
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nodes/mask_nodes.py
View File

@ -361,7 +361,7 @@ class CreateFluidMask:
return { return {
"required": { "required": {
"invert": ("BOOLEAN", {"default": False}), "invert": ("BOOLEAN", {"default": False}),
"frames": ("INT", {"default": 0,"min": 0, "max": 255, "step": 1}), "frames": ("INT", {"default": 1,"min": 1, "max": 4096, "step": 1}),
"width": ("INT", {"default": 256,"min": 16, "max": 4096, "step": 1}), "width": ("INT", {"default": 256,"min": 16, "max": 4096, "step": 1}),
"height": ("INT", {"default": 256,"min": 16, "max": 4096, "step": 1}), "height": ("INT", {"default": 256,"min": 16, "max": 4096, "step": 1}),
"inflow_count": ("INT", {"default": 3,"min": 0, "max": 255, "step": 1}), "inflow_count": ("INT", {"default": 3,"min": 0, "max": 255, "step": 1}),
@ -374,6 +374,9 @@ class CreateFluidMask:
#using code from https://github.com/GregTJ/stable-fluids #using code from https://github.com/GregTJ/stable-fluids
def createfluidmask(self, frames, width, height, invert, inflow_count, inflow_velocity, inflow_radius, inflow_padding, inflow_duration): def createfluidmask(self, frames, width, height, invert, inflow_count, inflow_velocity, inflow_radius, inflow_padding, inflow_duration):
from ..utility.fluid import Fluid from ..utility.fluid import Fluid
try:
from scipy.special import erf
except:
from scipy.spatial import erf from scipy.spatial import erf
out = [] out = []
masks = [] masks = []