From 108106df5113e4cbc4d24ad04ea635351275c09d Mon Sep 17 00:00:00 2001
From: wailovet <wailovet@163.com>
Date: Fri, 28 Jun 2024 00:24:25 +0800
Subject: [PATCH] 'update'

---
 mz_train_tools_core_HYDiT.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/mz_train_tools_core_HYDiT.py b/mz_train_tools_core_HYDiT.py
index 936a876..3cce23e 100644
--- a/mz_train_tools_core_HYDiT.py
+++ b/mz_train_tools_core_HYDiT.py
@@ -318,6 +318,13 @@ def check_required():
             sys.executable, "-m", "pip", "install", "diffusers"
         ], check=True)
     
+    try:
+        import transformers
+    except Exception as e:
+        subprocess.run([
+            sys.executable, "-m", "pip", "install", "transformers"
+        ], check=True)
+    
 
 
 def MZ_HYDiTTrain_call(args={}):