zxdu20's picture
Fix attention score on mps
cde457b