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