From cd2485f28d48e51b5304ffeef06df4cd944f26c7 Mon Sep 17 00:00:00 2001 From: marko1616 Date: Thu, 26 Sep 2024 11:14:09 -0400 Subject: [PATCH] Linter. Former-commit-id: 719d124f65ebb18ba0a1212751da9909160fb6f1 --- src/llamafactory/model/model_utils/visual.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/llamafactory/model/model_utils/visual.py b/src/llamafactory/model/model_utils/visual.py index 440ea8ce..5f983d9b 100644 --- a/src/llamafactory/model/model_utils/visual.py +++ b/src/llamafactory/model/model_utils/visual.py @@ -160,7 +160,9 @@ def get_image_seqlen(config: "PretrainedConfig") -> int: else: image_seqlen = -1 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