mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
Fix warn message (#163578)
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:
committed by
PyTorch MergeBot
parent
6b5ad5f211
commit
f3f67ff43a
@ -1413,7 +1413,7 @@ def flex_attention(
|
||||
elif return_lse and return_aux is None:
|
||||
_warn_once(
|
||||
"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.",
|
||||
category=FutureWarning,
|
||||
)
|
||||
|
Reference in New Issue
Block a user