frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix warn message (#163578)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/163578
Approved by: https://github.com/albanD, https://github.com/Skylion007, https://github.com/atalman, https://github.com/v0i0
This commit is contained in:
drisspg
2025-09-22 22:25:00 +00:00
committed by PyTorch MergeBot
parent 6b5ad5f211
commit f3f67ff43a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/nn/attention/flex_attention.py
View File

@ -1413,7 +1413,7 @@ def flex_attention(
elif return_lse and return_aux is None: elif return_lse and return_aux is None:
_warn_once( _warn_once(
"deprecated_return_lse", "deprecated_return_lse",
"return_lse is deprecated and will be removed in v2.7. " "return_lse is deprecated and will be removed in v2.10. "
"Please use return_aux=AuxRequest(lse=True) instead.", "Please use return_aux=AuxRequest(lse=True) instead.",
category=FutureWarning, category=FutureWarning,
) )