From 1a423fc953c8ab52d2a34c5672f3ec2a98b1334d Mon Sep 17 00:00:00 2001
From: wailovet <wailovet@163.com>
Date: Fri, 28 Jun 2024 00:42:20 +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..bc0b9bd 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", "protobuf"
     ]
     for package in packages:
         try: