Update modeling_reasonir_8b.py
Browse files- modeling_reasonir_8b.py +1 -1
modeling_reasonir_8b.py
CHANGED
@@ -997,7 +997,7 @@ class LlamaModel(LlamaPreTrainedModel):
|
|
997 |
output_hidden_states: Optional[bool] = None,
|
998 |
return_dict: Optional[bool] = None,
|
999 |
cache_position: Optional[torch.LongTensor] = None,
|
1000 |
-
is_causal: Optional[bool] =
|
1001 |
) -> Union[Tuple, BaseModelOutputWithPast]:
|
1002 |
output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions
|
1003 |
output_hidden_states = (
|
|
|
997 |
output_hidden_states: Optional[bool] = None,
|
998 |
return_dict: Optional[bool] = None,
|
999 |
cache_position: Optional[torch.LongTensor] = None,
|
1000 |
+
is_causal: Optional[bool] = False,
|
1001 |
) -> Union[Tuple, BaseModelOutputWithPast]:
|
1002 |
output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions
|
1003 |
output_hidden_states = (
|