Fix for RuntimeError: FlashAttention only support fp16 and bf16 data type during fine tuning. 5b7216f verified moidhassan commited on Sep 30, 2024