nielsr HF staff commited on
Commit
7463d6f
1 Parent(s): e29dc3b

Update modeling_cogvlm.py

Browse files
Files changed (1) hide show
  1. 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(