huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 12:24:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: improve typing

Browse Source
This commit is contained in:
drbh 2025-01-07 22:06:54 +00:00
parent 765ca78014
commit df504e9f1e
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
server/text_generation_server/models/vlm_causal_lm.py
View File

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