huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-06-19 07:42:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

nit

Browse Source
This commit is contained in:
Mohit Sharma 2025-04-22 14:41:20 +00:00
parent f1da19df41
commit dd91b60998
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -204,7 +204,8 @@ def preprocess_image(config, img):
img = img.resize((img.width * 2, img.height * 2))
elif model_type == "paligemma":
img = img.convert("RGB")
elif model_type not in {"llava_next", "gemma3", "llama4"}:
if model_type in {"llava_next", "gemma3", "llama4"}:
img = [img]
return img