diff --git a/torch/nn/attention/flex_attention.py b/torch/nn/attention/flex_attention.py index 7d631a7f8a3a..5c4c8184a06f 100644 --- a/torch/nn/attention/flex_attention.py +++ b/torch/nn/attention/flex_attention.py @@ -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, )