mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-09-09 19:34:53 +00:00
minor fix
Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>
This commit is contained in:
parent
8c988de92c
commit
800281113f
@ -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:
|
||||
|
Loading…
Reference in New Issue
Block a user