zyznull commited on
Commit
2c527fb
1 Parent(s): e79fe7d

Update modeling_qwen.py

Browse files
Files changed (1) hide show
  1. 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 = True,
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.