Difference in return sequence for Phi3 model

The Flash Attention warning is observed in both cases, but let me see if i can avoid it