Properly propagate `model_inputs`

modeling_plamo.py

@@ -1663,6 +1663,9 @@ class Plamo2ForCausalLM(Plamo2PreTrainedModel):
                 "position_ids": position_ids,
                 "past_key_values": past_key_values,
                 "use_cache": kwargs.get("use_cache"),
+                "output_attentions": kwargs.get("output_attentions"),
+                "output_hidden_states": kwargs.get("output_hidden_states"),
+                "logits_to_keep": kwargs.get("logits_to_keep"),
                 "attention_mask": attention_mask,
                 "image_features": image_features,
             }