From a65a9b11d300cc3f9b73efaf56b455720a32c93f Mon Sep 17 00:00:00 2001 From: ydshieh Date: Tue, 17 Dec 2024 17:49:49 +0100 Subject: [PATCH] temp --- src/transformers/models/kosmos2_5/modeling_kosmos2_5.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/kosmos2_5/modeling_kosmos2_5.py b/src/transformers/models/kosmos2_5/modeling_kosmos2_5.py index f67ba2d377ece4..f7d77e5961dc9d 100644 --- a/src/transformers/models/kosmos2_5/modeling_kosmos2_5.py +++ b/src/transformers/models/kosmos2_5/modeling_kosmos2_5.py @@ -1848,7 +1848,7 @@ def forward( vision_model_output = None projection_attentions = None if image_embeds is None: - if flattened_patches is not None and past_key_values is None: + if flattened_patches is not None: vision_model_output = self.vision_model( flattened_patches=flattened_patches, output_attentions=output_attentions, @@ -2161,7 +2161,7 @@ def forward( projection_attentions = None if image_embeds is None: - if flattened_patches is not None and past_key_values is None: + if flattened_patches is not None: vision_model_output = self.vision_model( flattened_patches=flattened_patches, output_attentions=output_attentions,