shmuli commited on
Commit
bfe13d5
·
verified ·
1 Parent(s): 4cfdf94

Update modeling_decilm.py

Browse files
Files changed (1) hide show
  1. modeling_decilm.py +1 -0
modeling_decilm.py CHANGED
@@ -1150,6 +1150,7 @@ class DeciLMForCausalLM(DeciLMPreTrainedModel, GenerationMixin):
1150
  output_hidden_states: Optional[bool] = None,
1151
  return_dict: Optional[bool] = None,
1152
  cache_position: Optional[torch.LongTensor] = None,
 
1153
  ) -> Union[Tuple, CausalLMOutputWithPast]:
1154
  r"""
1155
  Args:
 
1150
  output_hidden_states: Optional[bool] = None,
1151
  return_dict: Optional[bool] = None,
1152
  cache_position: Optional[torch.LongTensor] = None,
1153
+ **kwargs,
1154
  ) -> Union[Tuple, CausalLMOutputWithPast]:
1155
  r"""
1156
  Args: