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

Linter.

Browse Source
This commit is contained in:
marko1616
2024-09-26 11:14:09 -04:00
committed by hiyouga
parent 8372c5e377
commit b1e43e56db
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/llamafactory/model/model_utils/visual.py
View File

@@ -160,7 +160,9 @@ def get_image_seqlen(config: "PretrainedConfig") -> int:
else: else:
image_seqlen = -1 image_seqlen = -1
elif model_type == "mllama": elif model_type == "mllama":
image_seqlen = ((config.vision_config.image_size // config.vision_config.patch_size) ** 2 + 1) * config.vision_config.max_num_tiles image_seqlen = (
(config.vision_config.image_size // config.vision_config.patch_size) ** 2 + 1
) * config.vision_config.max_num_tiles
return image_seqlen return image_seqlen