Update modeling_cogvlm.py
Browse files- modeling_cogvlm.py +1 -0
modeling_cogvlm.py
CHANGED
@@ -745,6 +745,7 @@ class CogVLMForCausalLM(CogVLMPreTrainedModel):
|
|
745 |
model_kwargs: Dict[str, Any],
|
746 |
is_encoder_decoder: bool = False,
|
747 |
standardize_cache_format: bool = False,
|
|
|
748 |
) -> Dict[str, Any]:
|
749 |
# update past_key_values
|
750 |
model_kwargs["past_key_values"] = self._extract_past_from_model_output(
|
|
|
745 |
model_kwargs: Dict[str, Any],
|
746 |
is_encoder_decoder: bool = False,
|
747 |
standardize_cache_format: bool = False,
|
748 |
+
model_inputs: Optional[Dict[str, Any]] = None,
|
749 |
) -> Dict[str, Any]:
|
750 |
# update past_key_values
|
751 |
model_kwargs["past_key_values"] = self._extract_past_from_model_output(
|