Skip to content

Commit 94386ac

Browse files
committed
Fix parameter name
1 parent 6671cbe commit 94386ac

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

src/nets/common_former.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ def forward(self, x):
4444

4545
use_dropout = 0.0 if not self.training else self.dropout
4646

47-
with sdpa_kernel(self.sdp_backends, set_priority_order=True):
47+
with sdpa_kernel(self.sdp_backends, set_priority=True):
4848
context_vec = nn.functional.scaled_dot_product_attention(
4949
queries,
5050
keys,

src/nets/conformer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ def forward(self, x):
7979

8080
use_dropout = 0.0 if not self.training else self.dropout
8181

82-
with sdpa_kernel(self.sdp_backends, set_priority_order=True):
82+
with sdpa_kernel(self.sdp_backends, set_priority=True):
8383
context_vec = nn.functional.scaled_dot_product_attention(
8484
queries,
8585
keys,

0 commit comments

Comments
 (0)