From aa71571b773c5dc527b17219ec87828e4455b330 Mon Sep 17 00:00:00 2001 From: stephen_zhu Date: Fri, 8 Mar 2024 12:47:44 +0800 Subject: [PATCH] update --- 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 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__)