From 13dc46fffc855d2aa5131a6cdf8837339aa8467f Mon Sep 17 00:00:00 2001
From: City <125218114+city96@users.noreply.github.com>
Date: Tue, 21 May 2024 13:59:42 +0200
Subject: [PATCH] HunYuanDiT missing import

---
 HunYuanDiT/tenc.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/HunYuanDiT/tenc.py b/HunYuanDiT/tenc.py
index 7883982..311543b 100644
--- a/HunYuanDiT/tenc.py
+++ b/HunYuanDiT/tenc.py
@@ -4,6 +4,7 @@ import torch
 from transformers import AutoTokenizer, modeling_utils
 from transformers import T5Config, T5EncoderModel, BertConfig, BertModel
 
+from comfy import model_management
 import comfy.model_patcher
 import comfy.utils