diff --git a/src/llmtuner/model/patcher.py b/src/llmtuner/model/patcher.py
index a5d9e3b2..00e42147 100644
--- a/src/llmtuner/model/patcher.py
+++ b/src/llmtuner/model/patcher.py
@@ -24,7 +24,7 @@ if TYPE_CHECKING:
     from transformers import PretrainedConfig, PreTrainedTokenizer
     from trl import AutoModelForCausalLMWithValueHead
 
-    from ..hparams import ModelArguments
+    from ..hparams import ModelArguments,FinetuningArguments
 
 
 logger = get_logger(__name__)