diff --git a/server/text_generation_server/models/vlm_causal_lm.py b/server/text_generation_server/models/vlm_causal_lm.py index 41218d7c..ac46a717 100644 --- a/server/text_generation_server/models/vlm_causal_lm.py +++ b/server/text_generation_server/models/vlm_causal_lm.py @@ -387,7 +387,6 @@ class VlmCausalLMBatch(FlashCausalLMBatch): image_id = 0 for chunk in r.input_chunks.chunks: - chunk_type = chunk.WhichOneof("chunk") if chunk_type == "text": text = preprocess_text(config, chunk.text)