ros/LlamaFactory
1
0
Fork 0
mirror of https://github.com/hiyouga/LlamaFactory.git synced 2026-02-01 20:23:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix some

Browse Source 
Former-commit-id: cd5a1a8b9c6eb59d6e95f79573f60ad8668f1942
This commit is contained in:
fzc8578
2025-01-10 20:27:06 +08:00
parent 25d4889789
commit e63c2df0b1
5 changed files with 51 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/llamafactory/data/collator.py
View File

@@ -149,14 +149,14 @@ class MultiModalDataCollatorForSeq2Seq(DataCollatorForSeq2Seq):
features.update(mm_inputs)
if isinstance(features.get("pixel_values"), list): # for pixtral inputs
features = features.data # use default_collate() instead of BatchEncoding.to()
if "image_bound" in features: # for minicpmv inputs
if "image_bound" in features: # for minicpmv inputs
features = self.template.mm_plugin.pad_data(features)
new_features = {}
new_features.update({"data": features})
new_features.update(features)
features = new_features
return features