ColossalAI/applications/Colossal-LLaMA-2/colossal_llama2/utils
Yuanchen b397104438
[Colossal-Llama-2] Add finetuning Colossal-Llama-2 example (#4878)
* Add finetuning Colossal-Llama-2 example

* Add finetuning Colossal-Llama-2 example 2

* Add finetuning Colossal-Llama-2 example and support NEFTuning

* Add inference example and refine neftune

* Modify readme file

* update the imports

---------

Co-authored-by: Xu Yuanchen <yuanchen.xu00@gmail.com>
Co-authored-by: Camille Zhong <44392324+Camille7777@users.noreply.github.com>
2023-12-07 14:02:03 +08:00
..
__init__.py initial commit: add colossal llama 2 (#4784) 2023-09-24 23:12:26 +08:00
ckpt_io.py initial commit: add colossal llama 2 (#4784) 2023-09-24 23:12:26 +08:00
flash_attention_patch.py [format] applied code formatting on changed files in pull request 4908 (#4918) 2023-10-17 10:48:24 +08:00
froze.py initial commit: add colossal llama 2 (#4784) 2023-09-24 23:12:26 +08:00
neftune_patch.py [Colossal-Llama-2] Add finetuning Colossal-Llama-2 example (#4878) 2023-12-07 14:02:03 +08:00