From cd8c2219f2e6225edadb7efefb6b8f163fd8220d Mon Sep 17 00:00:00 2001
From: wailovet <wailovet@163.com>
Date: Wed, 19 Jun 2024 01:55:59 +0800
Subject: [PATCH] 'update'

---
 mz_train_tools_core.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/mz_train_tools_core.py b/mz_train_tools_core.py
index bafd03b..fa59103 100644
--- a/mz_train_tools_core.py
+++ b/mz_train_tools_core.py
@@ -231,7 +231,10 @@ def check_install():
     except ImportError:
         os.system(f"{sys.executable} -m pip install voluptuous")
 
-
+    try:
+        import diffusers
+    except ImportError:
+        os.system(f"{sys.executable} -m pip install diffusers")
 import logging