Update modeling_cogagent.py
Browse filesAdd missing required parameter, as per https://huggingface.co/THUDM/cogvlm-chat-hf/discussions/19
- modeling_cogagent.py +1 -0
modeling_cogagent.py
CHANGED
@@ -873,6 +873,7 @@ class CogAgentForCausalLM(CogAgentPreTrainedModel):
|
|
873 |
model_kwargs: Dict[str, Any],
|
874 |
is_encoder_decoder: bool = False,
|
875 |
standardize_cache_format: bool = False,
|
|
|
876 |
) -> Dict[str, Any]:
|
877 |
# update past_key_values
|
878 |
model_kwargs["past_key_values"] = self._extract_past_from_model_output(
|
|
|
873 |
model_kwargs: Dict[str, Any],
|
874 |
is_encoder_decoder: bool = False,
|
875 |
standardize_cache_format: bool = False,
|
876 |
+
model_inputs: Optional[Dict[str, Any]] = None,
|
877 |
) -> Dict[str, Any]:
|
878 |
# update past_key_values
|
879 |
model_kwargs["past_key_values"] = self._extract_past_from_model_output(
|