ColossalAI/applications/Colossal-LLaMA-2/colossal_llama2
Hongxin Liu a4cec1715b
[llama] add flash attn patch for npu (#5362)
2024-02-05 16:48:34 +08:00
..
dataset [llama] update training script (#5360) 2024-02-05 16:33:18 +08:00
model initial commit: add colossal llama 2 (#4784) 2023-09-24 23:12:26 +08:00
tokenizer [NFC] polish applications/Colossal-LLaMA-2/colossal_llama2/tokenizer/init_tokenizer.py code style (#5228) 2024-01-25 13:14:48 +08:00
utils [llama] add flash attn patch for npu (#5362) 2024-02-05 16:48:34 +08:00
__init__.py initial commit: add colossal llama 2 (#4784) 2023-09-24 23:12:26 +08:00