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