From dd91b609987bcb8963c49f8e18786753a9d9e2a6 Mon Sep 17 00:00:00 2001 From: Mohit Sharma Date: Tue, 22 Apr 2025 14:41:20 +0000 Subject: [PATCH] nit --- server/text_generation_server/models/vlm_causal_lm.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/server/text_generation_server/models/vlm_causal_lm.py b/server/text_generation_server/models/vlm_causal_lm.py index 17474eb4..953081a9 100644 --- a/server/text_generation_server/models/vlm_causal_lm.py +++ b/server/text_generation_server/models/vlm_causal_lm.py @@ -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