From f6ee34eb6e472c6c7cc2cc9786cbd217657456f0 Mon Sep 17 00:00:00 2001 From: huolongguo1O <121071167+huolongguo1O@users.noreply.github.com> Date: Tue, 17 Sep 2024 09:50:19 +0800 Subject: [PATCH] Update finetune.py MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 上一个commit在配置文件添加了freezeV选项,但在finetune.py中没有支持。https://github.com/THUDM/GLM-4/commit/1c2676415c213f534161da6ecc1d4b2de5be8caf#diff-a70f5de59d3f0aced07176ede208e2d2f420d7eafd2d9cf2478f93636160e538 暂时添加一句占位吧,虽然这个选项对纯文本微调没啥用 --- finetune_demo/finetune.py | 1 + 1 file changed, 1 insertion(+) diff --git a/finetune_demo/finetune.py b/finetune_demo/finetune.py index d5f0666..78c21fa 100644 --- a/finetune_demo/finetune.py +++ b/finetune_demo/finetune.py @@ -130,6 +130,7 @@ class FinetuningConfig(object): max_input_length: int max_output_length: int combine: bool + freezeV: bool training_args: Seq2SeqTrainingArguments = dc.field( default_factory=lambda: Seq2SeqTrainingArguments(output_dir='./output')