From 8125869ef75837863e06e02203c757844ddf92c3 Mon Sep 17 00:00:00 2001
From: wailovet <wailovet@163.com>
Date: Wed, 26 Jun 2024 11:54:29 +0800
Subject: [PATCH] 'update'

---
 .gitignore           | 1 +
 hook_kohya_ss_run.py | 6 ++++--
 2 files changed, 5 insertions(+), 2 deletions(-)

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