fix attention_mask missing

vit_gpt2/modeling_flax_vit_gpt2_lm.py

@@ -386,6 +386,7 @@ class FlaxViTGPT2LMPreTrainedModel(FlaxPreTrainedModel):
         return self.module.apply(
             {"params": params or self.params},
             pixel_values=jnp.array(pixel_values, dtype=jnp.float32),
+            attention_mask=attention_mask,
             decoder_input_ids=jnp.array(decoder_input_ids, dtype="i4"),
             decoder_attention_mask=jnp.array(decoder_attention_mask, dtype="i4"),
             decoder_position_ids=jnp.array(decoder_position_ids, dtype="i4"),