fixed validation error when using flash attention#2142

francesco-bertolotti wants to merge 2 commits intopytorch:mainpytorch/torchtitan:mainfrom

francesco-bertolotti:f14-flash-attn-valid-fixfrancesco-bertolotti/torchtitan:f14-flash-attn-valid-fixCopy head branch name to clipboard