fix: improve typing

huggingface · Jan 7, 2025 · df504e9 · df504e9
1 parent 765ca78
commit df504e9
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/server/text_generation_server/models/vlm_causal_lm.py b/server/text_generation_server/models/vlm_causal_lm.py
@@ -31,12 +31,12 @@
 
 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 = ""