fix pixel_attention_mask in batched call

modeling_idefics2.py

@@ -1506,8 +1506,10 @@ class Idefics2ForVisionText2Text(Idefics2PreTrainedModel):
         image_hidden_states = kwargs.get("image_hidden_states", None)
         if image_hidden_states is not None:
             pixel_values = None
+            pixel_attention_mask = None
         else:
             pixel_values = kwargs.get("pixel_values", None)
+            pixel_attention_mask = kwargs.get("pixel_attention_mask", None)
         model_inputs.update(
             {
                 "position_ids": position_ids,
@@ -1515,6 +1517,7 @@ class Idefics2ForVisionText2Text(Idefics2PreTrainedModel):
                 "use_cache": kwargs.get("use_cache"),
                 "attention_mask": attention_mask,
                 "pixel_values": pixel_values,
+                "pixel_attention_mask": pixel_attention_mask,
                 "image_hidden_states": image_hidden_states,
             }
         )