xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-01-09 07:14:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
kijai 2025-02-02 00:32:18 +02:00
parent e9a3170c72
commit a8aeef670b
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
nodes/nodes.py
View File

@ -2252,8 +2252,11 @@ class LeapfusionHunyuanI2V:
inp, timestep, c = args["input"], args["timestep"], args["c"]
current_step_index = (steps == timestep).nonzero().item()
current_percent = current_step_index / (len(steps) - 1)
if samples is not None and start_percent <= current_percent <= end_percent:
inp[:, :, [index], :, :] = samples[:, :, [0], :, :].to(inp)
if samples is not None:
if start_percent <= current_percent <= end_percent:
inp[:, :, [index], :, :] = samples[:, :, [0], :, :].to(inp)
else:
inp[:, :, [index], :, :] = torch.zeros(1)
return apply_model(inp, timestep, **c)
return unet_wrapper