Commit Graph

14 Commits

Author SHA1 Message Date
BUAADreamer
3dd5f9a874 add code for reading from multi files in one directory 2023-06-10 15:53:47 +08:00
hiyouga
17acf3a3eb add belle template 2023-06-07 12:30:11 +08:00
hiyouga
909af8f496 add prompt template class 2023-06-07 11:55:25 +08:00
hiyouga
44298c1235 tiny fix 2023-06-05 15:25:22 +08:00
hiyouga
eac9921e5c tiny fix 2023-06-04 12:55:40 +08:00
hiyouga
3b9eee8cd2 support QLoRA 2023-06-04 00:08:56 +08:00
hiyouga
de09ee1315 add ziya prompt template 2023-06-03 19:05:51 +08:00
hiyouga
771f454ff1 use low_cpu_mem_usage to speed up loading 2023-06-03 18:19:01 +08:00
hiyouga
dca27b4412 add logits processor 2023-06-03 16:34:54 +08:00
hiyouga
bd565af370 fix #1 2023-06-02 14:25:00 +08:00
hiyouga
740a5daf56 support BLOOM models 2023-05-31 16:54:06 +08:00
hiyouga
c0e5df92d6 fix checkpoint loading 2023-05-29 17:43:16 +08:00
hiyouga
ce71cc8b6d tiny fix 2023-05-29 09:42:29 +08:00
hiyouga
769c6ab56b Initial commit 2023-05-28 18:09:04 +08:00