From df504e9f1e274727c5fc62f512ec62e5cd800d29 Mon Sep 17 00:00:00 2001
From: drbh <david.richard.holtz@gmail.com>
Date: Tue, 7 Jan 2025 22:06:54 +0000
Subject: [PATCH] fix: improve typing

---
 .../text_generation_server/models/vlm_causal_lm.py   | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/server/text_generation_server/models/vlm_causal_lm.py b/server/text_generation_server/models/vlm_causal_lm.py
index daf5d063..0ff3c3fa 100644
--- a/server/text_generation_server/models/vlm_causal_lm.py
+++ b/server/text_generation_server/models/vlm_causal_lm.py
@@ -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 = ""