From 0b25f47bd9031a3f3415201e5c2ec16aaf8d9d0c Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Mon, 16 Dec 2024 19:42:01 -0500 Subject: [PATCH] Add some missing imports. --- comfy_extras/nodes_hunyuan.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/comfy_extras/nodes_hunyuan.py b/comfy_extras/nodes_hunyuan.py index 98f88daa..d6408269 100644 --- a/comfy_extras/nodes_hunyuan.py +++ b/comfy_extras/nodes_hunyuan.py @@ -1,3 +1,8 @@ +import nodes +import torch +import comfy.model_management + + class CLIPTextEncodeHunyuanDiT: @classmethod def INPUT_TYPES(s):