From b919203dc52ee4155d3b242c539a323e14bd1095 Mon Sep 17 00:00:00 2001 From: chenyang <15652611604@163.com> Date: Tue, 23 Jul 2024 18:36:03 +0800 Subject: [PATCH] This commit fixes a bug encountered when running the script to fine-tune the vision model. --- finetune_demo/finetune_vision.py | 1 + 1 file changed, 1 insertion(+) diff --git a/finetune_demo/finetune_vision.py b/finetune_demo/finetune_vision.py index c259d8a..71c7e31 100644 --- 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')