Commit Graph

9 Commits

Author SHA1 Message Date
hiyouga
4eb17bcf6c support distributed quantized training 2023-06-06 17:39:41 +08:00
hiyouga
dca27b4412 add logits processor 2023-06-03 16:34:54 +08:00
hiyouga
ed6161fa6a remove unused code 2023-06-03 00:10:54 +08:00
hiyouga
e3aaef7d4a fix layer norm name in PPO 2023-06-02 17:30:01 +08:00
hiyouga
50d9a20f81 alter rewards data type 2023-06-02 14:19:51 +08:00
hiyouga
740a5daf56 support BLOOM models 2023-05-31 16:54:06 +08:00
hiyouga
166c837b95 tiny fix 2023-05-28 21:48:33 +08:00
hiyouga
0c9fda01e3 use fp16 model, add logcallback 2023-05-28 21:30:28 +08:00
hiyouga
769c6ab56b Initial commit 2023-05-28 18:09:04 +08:00