Update model_minicpmv.py for latest compatibility (#174)

This commit is contained in:
qianyu chen
2024-05-29 16:03:59 +08:00
committed by GitHub
parent e88df8f04d
commit e75512cb9a
3 changed files with 31 additions and 3 deletions

View File

@@ -42,7 +42,7 @@ torchrun $DISTRIBUTED_ARGS finetune.py \
--output_dir output/output_minicpmv2 \
--logging_dir output/output_minicpmv2 \
--logging_strategy "steps" \
--per_device_train_batch_size 2 \
--per_device_train_batch_size 1 \
--per_device_eval_batch_size 1 \
--gradient_accumulation_steps 1 \
--evaluation_strategy "steps" \