diff --git a/src/llmtuner/model/patcher.py b/src/llmtuner/model/patcher.py
index c537a6d2..3b6f000a 100644
--- a/src/llmtuner/model/patcher.py
+++ b/src/llmtuner/model/patcher.py
@@ -286,7 +286,7 @@ def patch_model(
 
     if getattr(model.config, "model_type", None) == "mixtral" and is_deepspeed_zero3_enabled():
         require_version("deepspeed>=0.13.0", "To fix: pip install deepspeed>=0.13.0")
-        from deepspeed.utils import set_z3_leaf_modules
+        from deepspeed.utils import set_z3_leaf_modules  # type: ignore
         from transformers.models.mixtral.modeling_mixtral import MixtralSparseMoeBlock
         set_z3_leaf_modules(model, [MixtralSparseMoeBlock])