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

Linter.

Browse Source 
Former-commit-id: 719d124f65ebb18ba0a1212751da9909160fb6f1
This commit is contained in:
marko1616 2024-09-26 11:14:09 -04:00 committed by hiyouga
parent 918a367378
commit cd2485f28d
1 changed files with 3 additions and 1 deletions
Show Stats 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