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: