From 800281113f59de869f8d539a3a559308aa34da19 Mon Sep 17 00:00:00 2001 From: "Wang, Yi A" Date: Thu, 26 Jun 2025 22:19:38 -0700 Subject: [PATCH] minor fix Signed-off-by: Wang, Yi A --- .../server/text_generation_server/models/mllama_causal_lm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backends/gaudi/server/text_generation_server/models/mllama_causal_lm.py b/backends/gaudi/server/text_generation_server/models/mllama_causal_lm.py index 71ff1b55..b1f5232c 100644 --- a/backends/gaudi/server/text_generation_server/models/mllama_causal_lm.py +++ b/backends/gaudi/server/text_generation_server/models/mllama_causal_lm.py @@ -226,7 +226,7 @@ def generate_cross_attention_states( cross_attention_states, image_indices, input_lengths, pad_seq_len, prefilling ): if cross_attention_states is None: - return None, None, None + return None, None indices_list = [] if prefilling: for i in image_indices: