From 81d44d641454a27f15eadf91b1f909bf563ce103 Mon Sep 17 00:00:00 2001
From: wailovet <wailovet@163.com>
Date: Sun, 30 Jun 2024 20:22:01 +0800
Subject: [PATCH] 'update'

---
 mz_train_tools_core_HYDiT.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/mz_train_tools_core_HYDiT.py b/mz_train_tools_core_HYDiT.py
index abeb492..218fe66 100644
--- a/mz_train_tools_core_HYDiT.py
+++ b/mz_train_tools_core_HYDiT.py
@@ -851,7 +851,16 @@ def MZ_HYDiTSimpleT2I_call(args={}):
     if HYDiT_tool_dir not in sys.path:
         sys.path.append(HYDiT_tool_dir)
 
-    import hydit.inference
+
+
+
+
+    try:
+        import hydit
+    except ImportError:
+        sys.path.append(HYDiT_tool_dir)
+    
+    
     from hydit.inference import SAMPLER_FACTORY, HUNYUAN_DIT_CONFIG, get_fill_resize_and_crop, get_2d_rotary_pos_embed
     from hydit.modules.text_encoder import MT5Embedder
     from hydit.utils.tools import set_seeds