hiyouga f326178f89 support FlashAttention2 ...

Former-commit-id: 23e56c5554b948d4f08ad87849b261eafd2c7890

2023-09-10 20:43:56 +08:00

..

models

support FlashAttention2

2023-09-10 20:43:56 +08:00

__init__.py

modity code structure

2023-07-15 16:54:28 +08:00

callbacks.py

support DPO training (2305.18290)

2023-08-11 03:02:53 +08:00

constants.py

update baichuan2 template

2023-09-06 21:43:06 +08:00

logging.py

support rope scaling, fix #475 #476 #478

2023-08-12 20:46:27 +08:00

misc.py

support lora target auto find

2023-09-09 15:38:37 +08:00

ploting.py

release v0.1.0

2023-07-18 00:18:25 +08:00

save_and_load.py

support lora target auto find

2023-09-09 15:38:37 +08:00

template.py

support FlashAttention2

2023-09-10 20:43:56 +08:00