Just add FLASH_ATTENTION_TRITON_AMD_ENABLE=TRUE env variable before the command. It will go AMD route ...