15 Commits

Author SHA1 Message Date
hiyouga
4724ae3492 support RM metrics, add generating Args
Former-commit-id: c461c6190bc124e98dde7f3cf96a59ce40b26fb0
2023-06-12 15:48:48 +08:00
hiyouga
b9feb82e4e add prompt template class
Former-commit-id: 3d7e3a38d00aa5d9664824093043951af8c3f707
2023-06-07 11:55:25 +08:00
hiyouga
3da427a665 fix inference, add prompt template
Former-commit-id: 3940e50c71472b210bbc1b01248bf85a191c4065
2023-06-07 10:52:35 +08:00
hiyouga
ac6f50dedf add API demo from #1
Former-commit-id: c955edcef168da44257c5b50d7bc59266d909782
2023-06-05 21:32:18 +08:00
hiyouga
a409e1f42c fix bug in web demo
Former-commit-id: 01d6d7a910b9845a0ea38632661ce813e5cfe3a2
2023-06-05 17:58:29 +08:00
hiyouga
f9c51a8340 implement stream generating
Former-commit-id: 6cc9535975d823ffef7e1686749b69b40347a8ec
2023-06-05 16:43:44 +08:00
hiyouga
2308d5a179 fix int8 inference
Former-commit-id: d05202943e9634526f96d189288f67852d3d1c40
2023-06-03 23:22:05 +08:00
hiyouga
7d6542115c reduce repetition penalty
Former-commit-id: 0381d93fc10ea5346724fe6295caa565a8eb4f61
2023-06-03 21:57:39 +08:00
hiyouga
fa850ae6e5 add ziya prompt template
Former-commit-id: 321e44ac54a91260cf00a4caa1991708814473fc
2023-06-03 19:05:51 +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
1512711ca2 fix bug at inference
Former-commit-id: df9b41af4401006b8040eb53c44dd290b604e0eb
2023-05-31 18:11:53 +08:00
hiyouga
693c049eac support BLOOM models
Former-commit-id: 1314b6ea39a01aa8ac325e1d875ac013d43aec45
2023-05-31 16:54:06 +08:00
hiyouga
304be6dc28 fix checkpoint loading
Former-commit-id: d31aa5c2c0bcb6a4ef4a62e21693548dd9acaae6
2023-05-29 17:43:16 +08:00
hiyouga
17024ebc1a Initial commit
Former-commit-id: 5ca8e1d63727e7bcb8cab16542c763c47e48184a
2023-05-28 18:09:04 +08:00