From 60a042cc16b6e8d7eaf90dfc196f5471e249d1cd Mon Sep 17 00:00:00 2001 From: hiyouga Date: Mon, 22 Jan 2024 23:32:01 +0800 Subject: [PATCH] add hint Former-commit-id: e4ba1deedfc366c984f0095b250ed42a57c3dfba --- src/llmtuner/model/patcher.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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])