fix: improve typing

This commit is contained in:
drbh 2025-01-07 22:06:54 +00:00
parent 765ca78014
commit df504e9f1e

View File

@ -31,12 +31,12 @@ IDEFICS3_GLOBAL_IMG_TOKEN = "<global-img>"
def get_image_prompt_string(
*,
image_seq_len,
image_rows,
image_cols,
fake_token_around_image,
image_token,
global_img_token,
image_seq_len: int,
image_rows: int,
image_cols: int,
fake_token_around_image: str,
image_token: str,
global_img_token: str,
):
"""Prompt with expanded image tokens for when the image is split into patches."""
text_split_images = ""