We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 5713ef9 + 3136732 commit 5a016d5Copy full SHA for 5a016d5
examples/models/llama/source_transformation/sdpa.py
@@ -47,7 +47,6 @@ def forward(
47
torch._check_is_size(start_pos)
48
torch._check(start_pos < self.max_context_len)
49
seq_length = q.size(2)
50
- # pyre-ignore: Incompatible parameter type [6]
51
mask = mask.narrow(0, start_pos, seq_length)
52
else:
53
mask = mask[input_pos]
0 commit comments