xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-06-05 05:17:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
kijai 2025-02-01 23:58:46 +02:00
parent 23103dffde
commit e11695a305
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nodes/nodes.py
View File

@ -2233,6 +2233,7 @@ class LeapfusionHunyuanI2V:
"model": ("MODEL",), "model": ("MODEL",),
"latent": ("LATENT",), "latent": ("LATENT",),
"index": ("INT", {"default": 0, "min": -1, "max": 1000, "step": 1,"tooltip": "The index of the latent to be replaced. 0 for first frame and -1 for last"}), "index": ("INT", {"default": 0, "min": -1, "max": 1000, "step": 1,"tooltip": "The index of the latent to be replaced. 0 for first frame and -1 for last"}),
"strength": ("FLOAT", {"default": 1.0, "min": -10.0, "max": 10.0, "step": 0.001}),
} }
} }
@ -2241,7 +2242,7 @@ class LeapfusionHunyuanI2V:
CATEGORY = "KJNodes/experimental" CATEGORY = "KJNodes/experimental"
def patch(self, model, latent, index): def patch(self, model, latent, index, strength):
def outer_wrapper(samples, index): def outer_wrapper(samples, index):
def unet_wrapper(apply_model, args): def unet_wrapper(apply_model, args):
@ -2251,7 +2252,7 @@ class LeapfusionHunyuanI2V:
return apply_model(inp, timestep, **c) return apply_model(inp, timestep, **c)
return unet_wrapper return unet_wrapper
samples = latent["samples"] * 0.476986 samples = latent["samples"] * 0.476986 * strength
m = model.clone() m = model.clone()
m.set_model_unet_function_wrapper(outer_wrapper(samples, index)) m.set_model_unet_function_wrapper(outer_wrapper(samples, index))