423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-15 08:08:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix bug

Browse Source 
Former-commit-id: 365e6df71509569f59c40743c115f1a4b945ef0f
This commit is contained in:
hoshi-hiyouga 2024-08-30 02:05:26 +08:00 committed by GitHub
parent 5b4b60cfb5
commit 7466fd4387
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/data/processors/supervised.py
View File

@ -50,7 +50,7 @@ def _encode_supervised_example(
train_on_prompt: bool, train_on_prompt: bool,
mask_history: bool, mask_history: bool,
) -> Tuple[List[int], List[int]]: ) -> Tuple[List[int], List[int]]:
if "image_grid_thw" in processor.model_input_names: # qwen2_vl models if processor is not None and "image_grid_thw" in processor.model_input_names: # qwen2_vl models
image_processor = getattr(processor, "image_processor") image_processor = getattr(processor, "image_processor")
merge_length = image_processor.merge_size**2 merge_length = image_processor.merge_size**2
if len(images) > 0: if len(images) > 0: