momergul commited on
Commit
4be5292
·
verified ·
1 Parent(s): 5d8f609

Upload modeling_git.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. modeling_git.py +1 -0
modeling_git.py CHANGED
@@ -53,6 +53,7 @@ class GitForCausalLM(modeling_git.GitForCausalLM):
53
  output_attentions: Optional[bool] = None,
54
  output_hidden_states: Optional[bool] = None,
55
  return_dict: Optional[bool] = None,
 
56
  ) -> Union[Tuple[torch.Tensor], modeling_git.CausalLMOutputWithPast]:
57
  return_dict = return_dict if return_dict is not None else self.config.use_return_dict
58
  if labels is not None:
 
53
  output_attentions: Optional[bool] = None,
54
  output_hidden_states: Optional[bool] = None,
55
  return_dict: Optional[bool] = None,
56
+ **kwargs,
57
  ) -> Union[Tuple[torch.Tensor], modeling_git.CausalLMOutputWithPast]:
58
  return_dict = return_dict if return_dict is not None else self.config.use_return_dict
59
  if labels is not None: