You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/examples/language/llama2/attn.py

Symbolic link
1 line
85 B

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