wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-30 04:12:37 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

Rename model input

Browse Source
This commit is contained in:
kijai 2026-04-22 12:21:11 +03:00
parent fd8bfce72a
commit 764e835533
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
comfy_extras/nodes_frame_interpolation.py
View File

@ -81,7 +81,7 @@ class FrameInterpolate(io.ComfyNode):
category="image/video",
search_aliases=["rife", "film", "frame interpolation", "slow motion", "interpolate frames", "vfi"],
inputs=[
FrameInterpolationModel.Input("model"),
FrameInterpolationModel.Input("frame_interpolation_model"),
io.Image.Input("images"),
io.Int.Input("multiplier", default=2, min=2, max=16),
],
@ -91,17 +91,17 @@ class FrameInterpolate(io.ComfyNode):
)
@classmethod
def execute(cls, model, images, multiplier) -> io.NodeOutput:
def execute(cls, frame_interpolation_model, images, multiplier) -> io.NodeOutput:
offload_device = model_management.intermediate_device()
num_frames = images.shape[0]
if num_frames < 2 or multiplier < 2:
return io.NodeOutput(images)
model_management.load_model_gpu(model)
device = model.load_device
dtype = model.model_dtype()
inference_model = model.model
model_management.load_model_gpu(frame_interpolation_model)
device = frame_interpolation_model.load_device
dtype = frame_interpolation_model.model_dtype()
inference_model = frame_interpolation_model.model
# Free VRAM for inference activations (model weights + ~20x a single frame's worth)
H, W = images.shape[1], images.shape[2]