Skip to content

Commit 3abee62

Browse files
pytorchbotdrisspg
andauthored
Fix warn message (pytorch#164367)
Fix warn message (pytorch#163578) Pull Request resolved: pytorch#163578 Approved by: https://github.com/albanD, https://github.com/Skylion007, https://github.com/atalman, https://github.com/v0i0 (cherry picked from commit f3f67ff) Co-authored-by: drisspg <[email protected]>
1 parent f227c88 commit 3abee62

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch/nn/attention/flex_attention.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1612,7 +1612,7 @@ def score_mod(
16121612
elif return_lse and return_aux is None:
16131613
_warn_once(
16141614
"deprecated_return_lse",
1615-
"return_lse is deprecated and will be removed in v2.7. "
1615+
"return_lse is deprecated and will be removed in v2.10. "
16161616
"Please use return_aux=AuxRequest(lse=True) instead.",
16171617
category=FutureWarning,
16181618
)

0 commit comments

Comments
 (0)