From 23fc0c863eff7c5513dbdb1de59ebb6fd5bcab55 Mon Sep 17 00:00:00 2001 From: marko1616 Date: Thu, 26 Sep 2024 11:14:09 -0400 Subject: [PATCH] Linter. Former-commit-id: b1e43e56db4d898c0d6d27b703b82a1eb9af70c2 --- 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