From 931ff90c79c501905e6d8e74c7cc9ca4cf6af050 Mon Sep 17 00:00:00 2001
From: wailovet <wailovet@163.com>
Date: Wed, 19 Jun 2024 02:00:46 +0800
Subject: [PATCH] 'update'

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

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