wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-23 17:02:38 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

Remove torch.compile for now

Browse Source
This commit is contained in:
kijai 2026-04-13 23:08:15 +03:00
parent 1415828115
commit fe32b376c3
1 changed files with 1 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
comfy_extras/nodes_frame_interpolation.py
View File

@ -84,8 +84,6 @@ class FrameInterpolate(io.ComfyNode):
FrameInterpolationModel.Input("model"), FrameInterpolationModel.Input("model"),
io.Image.Input("images"), io.Image.Input("images"),
io.Int.Input("multiplier", default=2, min=2, max=16), io.Int.Input("multiplier", default=2, min=2, max=16),
io.Boolean.Input("torch_compile", default=False, optional=True, advanced=True,
tooltip="Requires triton. Compile model submodules for potential speed increase. Adds warmup on first run, recompiles on resolution change."),
], ],
outputs=[ outputs=[
io.Image.Output(), io.Image.Output(),
@ -93,7 +91,7 @@ class FrameInterpolate(io.ComfyNode):
) )
@classmethod @classmethod
def execute(cls, model, images, multiplier, torch_compile=False) -> io.NodeOutput: def execute(cls, model, images, multiplier) -> io.NodeOutput:
offload_device = model_management.intermediate_device() offload_device = model_management.intermediate_device()
num_frames = images.shape[0] num_frames = images.shape[0]
@ -119,15 +117,6 @@ class FrameInterpolate(io.ComfyNode):
frame = pad_to_patch_size(frame, (align, align), padding_mode="reflect") frame = pad_to_patch_size(frame, (align, align), padding_mode="reflect")
return frame return frame
if torch_compile:
for name, child in inference_model.named_children():
if isinstance(child, (torch.nn.ModuleList, torch.nn.ModuleDict)):
continue
if not hasattr(child, "_compiled"):
compiled = torch.compile(child)
compiled._compiled = True
setattr(inference_model, name, compiled)
# Count total interpolation passes for progress bar # Count total interpolation passes for progress bar
total_pairs = num_frames - 1 total_pairs = num_frames - 1
num_interp = multiplier - 1 num_interp = multiplier - 1