hiyouga
|
c68f9ec3a9
|
fix int8 inference
Former-commit-id: fcf3506bef28504dd679c2210bdc84e5868e05fe
|
2023-06-03 21:17:47 +08:00 |
|
hiyouga
|
fa850ae6e5
|
add ziya prompt template
Former-commit-id: 321e44ac54a91260cf00a4caa1991708814473fc
|
2023-06-03 19:05:51 +08:00 |
|
hiyouga
|
5eef8d5d98
|
use low_cpu_mem_usage to speed up loading
Former-commit-id: 7891e4c200566a4a47088e93efd1fbebcb46528e
|
2023-06-03 18:19:01 +08:00 |
|
hiyouga
|
9b8b6623ac
|
add logits processor
Former-commit-id: f6f4b1554ae1e8849b437d705ffa34ce7ebd56bb
|
2023-06-03 16:34:54 +08:00 |
|
hiyouga
|
e9ab06678f
|
alter rewards data type
Former-commit-id: 3eb7eb2d37525da50fe401ab7c59532e6e1ef984
|
2023-06-02 14:19:51 +08:00 |
|
hiyouga
|
896dbfec16
|
fix possibly OOM error
Former-commit-id: 0d590dffb41b0e832d9f87d20a23bcd0acd983aa
|
2023-06-01 23:54:44 +08:00 |
|
hiyouga
|
693c049eac
|
support BLOOM models
Former-commit-id: 1314b6ea39a01aa8ac325e1d875ac013d43aec45
|
2023-05-31 16:54:06 +08:00 |
|
hiyouga
|
181c776b58
|
remove dummy code
Former-commit-id: e6bc89d280945bbf48281107145c40a41d7cbd56
|
2023-05-30 16:28:00 +08:00 |
|
hiyouga
|
33fee45217
|
add pre-training script
Former-commit-id: 935d58de2b3a2eadc4f0bed28c3ad7dee32e9fd5
|
2023-05-29 21:37:22 +08:00 |
|
hiyouga
|
304be6dc28
|
fix checkpoint loading
Former-commit-id: d31aa5c2c0bcb6a4ef4a62e21693548dd9acaae6
|
2023-05-29 17:43:16 +08:00 |
|
hiyouga
|
35d04a2c05
|
tiny fix
Former-commit-id: eae79707d31fd8be2cf4bee4d610557bbd49f6e7
|
2023-05-29 09:42:29 +08:00 |
|
hiyouga
|
1fc551e1be
|
use fp16 model, add logcallback
Former-commit-id: bea275d51338b49ce855eec0178e759607265e3d
|
2023-05-28 21:30:28 +08:00 |
|
hiyouga
|
17024ebc1a
|
Initial commit
Former-commit-id: 5ca8e1d63727e7bcb8cab16542c763c47e48184a
|
2023-05-28 18:09:04 +08:00 |
|