deepgeek/MiniCPM-o
2
0
Fork 0
mirror of https://github.com/OpenBMB/MiniCPM-V.git synced 2026-02-05 18:29:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
qianyu chen
2024-08-14 18:31:28 +08:00
committed by GitHub
parent 7842ec1228
commit 3b3b9331cb
4 changed files with 35 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
finetune/finetune.py
View File

@@ -108,6 +108,7 @@ def make_supervised_data_module(
patch_size=patch_size,
query_nums=query_nums,
batch_vision=batch_vision,
max_length=max_length,
)
if data_args.eval_data_path:
@@ -121,6 +122,7 @@ def make_supervised_data_module(
patch_size=patch_size,
query_nums=query_nums,
batch_vision=batch_vision,
max_length=max_length,
)
else:
eval_dataset = None
@@ -205,7 +207,6 @@ def train():
tokenizer = AutoTokenizer.from_pretrained(
model_args.model_name_or_path, trust_remote_code=True
)
tokenizer.model_max_length = training_args.model_max_length
if not training_args.tune_vision:
model.vpm.requires_grad_(False)