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 = (
|