ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

tiny fix

Browse Source 
Former-commit-id: 8b4f408da110d74285bae20bbd969013a979964b
This commit is contained in:
hiyouga
2024-09-02 01:33:22 +08:00
parent bae73e676c
commit f203a9d78e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/data/mm_plugin.py
View File

@@ -208,7 +208,7 @@ class Qwen2vlPlugin(BasePlugin):
) -> List[Dict[str, str]]: ) -> List[Dict[str, str]]:
image_processor: "BaseImageProcessor" = getattr(processor, "image_processor") image_processor: "BaseImageProcessor" = getattr(processor, "image_processor")
merge_length: int = getattr(image_processor, "merge_size") ** 2 merge_length: int = getattr(image_processor, "merge_size") ** 2
if len(images) > 0: if len(images) != 0:
image_grid_thw = _get_mm_inputs(images, processor)["image_grid_thw"] image_grid_thw = _get_mm_inputs(images, processor)["image_grid_thw"]
else: else:
image_grid_thw = [] image_grid_thw = []