efort
/
glm4
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

This commit is contained in:
chenyang 2024-07-23 18:36:03 +08:00
parent 7f96ee326e
commit b919203dc5
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
finetune_demo/finetune_vision.py
View File

@ -135,6 +135,7 @@ class FinetuningConfig(object):
max_input_length: int max_input_length: int
max_output_length: int max_output_length: int
combine: bool
training_args: Seq2SeqTrainingArguments = dc.field( training_args: Seq2SeqTrainingArguments = dc.field(
default_factory=lambda: Seq2SeqTrainingArguments(output_dir='./output') default_factory=lambda: Seq2SeqTrainingArguments(output_dir='./output')