diff --git a/comfy_extras/v3/nodes_torch_compile.py b/comfy_extras/v3/nodes_torch_compile.py index ed218174c..ec047ef9e 100644 --- a/comfy_extras/v3/nodes_torch_compile.py +++ b/comfy_extras/v3/nodes_torch_compile.py @@ -1,7 +1,7 @@ from __future__ import annotations -from comfy_api.torch_helpers import set_torch_compile_wrapper from comfy_api.latest import io +from comfy_api.torch_helpers import set_torch_compile_wrapper class TorchCompileModel(io.ComfyNode): diff --git a/comfy_extras/v3/nodes_video.py b/comfy_extras/v3/nodes_video.py index 004efe08f..0fb8b0f5b 100644 --- a/comfy_extras/v3/nodes_video.py +++ b/comfy_extras/v3/nodes_video.py @@ -11,8 +11,8 @@ import folder_paths from comfy.cli_args import args from comfy_api.input import AudioInput, ImageInput, VideoInput from comfy_api.input_impl import VideoFromComponents, VideoFromFile -from comfy_api.util import VideoCodec, VideoComponents, VideoContainer from comfy_api.latest import io, ui +from comfy_api.util import VideoCodec, VideoComponents, VideoContainer class CreateVideo(io.ComfyNode):