This commit fixes a bug encountered when running the script to fine-tune the vision model.

2024-07-23 18:36:03 +08:00 · 2024-07-23 18:36:03 +08:00 · b919203dc5
parent 7f96ee326e
commit b919203dc5
1 changed files with 1 additions and 0 deletions
--- a/finetune_demo/finetune_vision.py
+++ b/finetune_demo/finetune_vision.py
@ -135,6 +135,7 @@ class FinetuningConfig(object):

    max_input_length: int
    max_output_length: int
+    combine: bool

    training_args: Seq2SeqTrainingArguments = dc.field(
        default_factory=lambda: Seq2SeqTrainingArguments(output_dir='./output')