diff --git a/.gitignore b/.gitignore
index 5130811..03b2b0f 100644
--- a/.gitignore
+++ b/.gitignore
@@ -171,3 +171,4 @@ exclude.txt
 *.msgpack
 *.onnx_data
 configs/models_config/**/*.png
+HunyuanDiT
\ No newline at end of file
diff --git a/hook_kohya_ss_run.py b/hook_kohya_ss_run.py
index d0488e6..73a2209 100644
--- a/hook_kohya_ss_run.py
+++ b/hook_kohya_ss_run.py
@@ -31,8 +31,10 @@ import numpy as np
 import tempfile
 import safetensors.torch
 
-import hook_kohya_ss_utils
-
+try:
+    import hook_kohya_ss_utils
+except:
+    from . import hook_kohya_ss_utils
 
 other_config = {}
 original_save_model = None