fix FlaxViTGPT2LMModule return value

vit_gpt2/modeling_flax_vit_gpt2_lm.py

@@ -78,8 +78,8 @@ class FlaxViTGPT2LMModule(nn.Module):
 
         return FlaxSeq2SeqLMOutput(
             logits=decoder_outputs.logits,
-            decoder_hidden_states=decoder_outputs.decoder_hidden_states,
-            decoder_attentions=decoder_outputs.decoder_attentions,
+            decoder_hidden_states=decoder_outputs.hidden_states,
+            decoder_attentions=decoder_outputs.attentions,
             cross_attentions=decoder_outputs.cross_attentions,
             encoder_last_hidden_state=encoder_outputs.last_hidden_state,
             encoder_hidden_states=encoder_outputs.hidden_states,