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