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

Merge branch 'main' of https://github.com/OpenBMB/MiniCPM-V

Browse Source 
t push
This commit is contained in:
yiranyyu
2024-05-28 15:02:16 +08:00
parent faf3ba957f 5fdf2e0fd6
commit e62beb985f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
finetune/dataset.py
View File

@@ -147,7 +147,7 @@ def conversation_to_ids(conversation, tokenizer, llm_type=None):
else: else:
image_bound = [] image_bound = []
position_ids = torch.where(ids != 0, torch.arange(ids.size(0)), torch.tensor(0)).long() position_ids = torch.arange(ids.size(0)).long()
return { return {
"input_ids": ids, "input_ids": ids,
"target": target, "target": target,