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

Update nodes.py

Browse Source
This commit is contained in:
kijai 2025-02-01 22:15:13 +02:00
parent 6c4aa273c8
commit 81b7d71454
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
nodes/nodes.py
View File

@ -2232,6 +2232,7 @@ class LeapfusionHunyuanI2V:
"required": { "required": {
"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"}),
} }
} }
@ -2240,18 +2241,18 @@ class LeapfusionHunyuanI2V:
CATEGORY = "KJNodes/experimental" CATEGORY = "KJNodes/experimental"
def patch(self, model, latent): def patch(self, model, latent, index):
def outer_wrapper(samples): def outer_wrapper(samples, index):
def unet_wrapper(apply_model, args): def unet_wrapper(apply_model, args):
inp, timestep, c = args["input"], args["timestep"], args["c"] inp, timestep, c = args["input"], args["timestep"], args["c"]
if samples is not None: if samples is not None:
inp[:, :, [0], :, :] = samples[:, :, [0], :, :].to(inp) inp[:, :, [index], :, :] = samples[:, :, [0], :, :].to(inp)
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
m = model.clone() m = model.clone()
m.set_model_unet_function_wrapper(outer_wrapper(samples)) m.set_model_unet_function_wrapper(outer_wrapper(samples, index))
return (m,) return (m,)