Update modeling_qwen.py
Browse files- modeling_qwen.py +1 -1
modeling_qwen.py
CHANGED
@@ -646,7 +646,7 @@ class Qwen2SdpaAttention(Qwen2Attention):
|
|
646 |
past_key_value: Optional[Cache] = None,
|
647 |
output_attentions: bool = False,
|
648 |
use_cache: bool = False,
|
649 |
-
is_causal: bool =
|
650 |
) -> Tuple[torch.Tensor, Optional[torch.Tensor], Optional[Tuple[torch.Tensor]]]:
|
651 |
if output_attentions:
|
652 |
# TODO: Improve this warning with e.g. `model.config.attn_implementation = "manual"` once this is implemented.
|
|
|
646 |
past_key_value: Optional[Cache] = None,
|
647 |
output_attentions: bool = False,
|
648 |
use_cache: bool = False,
|
649 |
+
is_causal: bool = False,
|
650 |
) -> Tuple[torch.Tensor, Optional[torch.Tensor], Optional[Tuple[torch.Tensor]]]:
|
651 |
if output_attentions:
|
652 |
# TODO: Improve this warning with e.g. `model.config.attn_implementation = "manual"` once this is implemented.
|