Update modeling_internvl_chat.py

modeling_internvl_chat.py

@@ -1125,8 +1125,8 @@ class Qwen2Model(Qwen2PreTrainedModel):
             attention_mask, inputs_embeds, cache_position, past_key_values, output_attentions
         )
         #TODO Motify
-        if causal_mask.shape[-1] == causal_mask.shape[-2]:
-            causal_mask[:, :, -1, :-1] = causal_mask[:, :, -1, :-1] + causal_mask[:, :, 0, -1] 
+        #if causal_mask.shape[-1] == causal_mask.shape[-2]:
+        #    causal_mask[:, :, -1, :-1] = causal_mask[:, :, -1, :-1] + causal_mask[:, :, 0, -1] 
         
 
         hidden_states = inputs_embeds