Update modeling_cogagent.py

modeling_cogagent.py

@@ -873,6 +873,7 @@ class CogAgentForCausalLM(CogAgentPreTrainedModel):
             model_kwargs: Dict[str, Any],
             is_encoder_decoder: bool = False,
             standardize_cache_format: bool = False,
+            model_inputs: Optional[Dict[str, Any]] = None,
     ) -> Dict[str, Any]:
         # update past_key_values
         model_kwargs["past_key_values"] = self._extract_past_from_model_output(