From 50e5a6829b8326530d4db50d7bed06cb3ed0dad1 Mon Sep 17 00:00:00 2001
From: wailovet <wailovet@163.com>
Date: Fri, 28 Jun 2024 00:30:59 +0800
Subject: [PATCH] 'update'

---
 mz_train_tools_core_HYDiT.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mz_train_tools_core_HYDiT.py b/mz_train_tools_core_HYDiT.py
index bf5a457..62e3bb5 100644
--- a/mz_train_tools_core_HYDiT.py
+++ b/mz_train_tools_core_HYDiT.py
@@ -321,7 +321,7 @@ def check_required():
     packages = [
         "pandas", "pyarrow", "diffusers", "transformers",
         "timm", "peft", "accelerate", "loguru", "einops", "sentencepiece",
-        "polygraphy"
+        "polygraphy", "deepspeed"
     ]
     for package in packages:
         try: