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')