Upload modeling_git.py with huggingface_hub
Browse files- 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:
|