blewis-hir commited on
Commit
57d602b
·
verified ·
1 Parent(s): 277a89d

Fix crash on model forward with fix by @shmuli

Browse files
Files changed (1) hide show
  1. modeling_decilm.py +1 -0
modeling_decilm.py CHANGED
@@ -1159,6 +1159,7 @@ class DeciLMForCausalLM(DeciLMPreTrainedModel, GenerationMixin):
1159
  output_hidden_states: Optional[bool] = None,
1160
  return_dict: Optional[bool] = None,
1161
  cache_position: Optional[torch.LongTensor] = None,
 
1162
  ) -> Union[Tuple, CausalLMOutputWithPast]:
1163
  r"""
1164
  Args:
 
1159
  output_hidden_states: Optional[bool] = None,
1160
  return_dict: Optional[bool] = None,
1161
  cache_position: Optional[torch.LongTensor] = None,
1162
+ **kwargs,
1163
  ) -> Union[Tuple, CausalLMOutputWithPast]:
1164
  r"""
1165
  Args: