Merge pull request #563 from huolongguo1O/patch-1

解决finetune.py在加载配置文件时的错误
2024-10-16 17:36:19 +08:00 · 2024-10-16 17:36:19 +08:00 · 4e9b473d18
parent 5142bdb6e1 f6ee34eb6e
commit 4e9b473d18
1 changed files with 1 additions and 0 deletions
--- 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')