../../../applications/Colossal-LLaMA-2/colossal_llama2/utils/flash_attention_patch.py