drisspg
2025-09-22 22:25:00 +00:00
committed by PyTorch MergeBot
parent 6b5ad5f211
commit f3f67ff43a

View File

@ -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,
)