2759 Commits

Author SHA1 Message Date
hiyouga
3836aadacf support loading lora from hub
Former-commit-id: 0b34c962bc3368dca62b18ad6c27a0293c3affa5
2023-06-16 00:02:17 +08:00
hiyouga
194c5d2bee support baichuan model
Former-commit-id: d683042fbcb2ee43b9823262d0a65b64f4cb54cb
2023-06-15 16:02:01 +08:00
hiyouga
496846e819 fix bug in template vanilla
Former-commit-id: 9b51e44c95af116aec34e7b6495935420f7c6c27
2023-06-15 14:36:55 +08:00
hiyouga
c42562d7ae add BOS token in pre-training
Former-commit-id: c57cf5d4a46c57c6f698e5cfd0fd59cce703094d
2023-06-15 01:46:17 +08:00
hiyouga
aa1bb8a9a2 support multiturn training like FastChat
Former-commit-id: 629cafb1a09924e82d7ea1f9fba318d3f5593196
2023-06-14 22:27:39 +08:00
hiyouga
6f655e3916 fix loading valuehead
Former-commit-id: 7872375d7a0c1d8826206631f6717a91ec49f1b3
2023-06-13 11:13:06 +08:00
hiyouga
6828f07d54 fix generating args
Former-commit-id: 52805a8441bd7b324bd89489de60f18f103c8e4c
2023-06-13 01:33:56 +08:00
hiyouga
4724ae3492 support RM metrics, add generating Args
Former-commit-id: c461c6190bc124e98dde7f3cf96a59ce40b26fb0
2023-06-12 15:48:48 +08:00
hoshi-hiyouga
4c5cad9722 Merge pull request #26 from BUAADreamer/main
add code for reading from multi files in one directory

Former-commit-id: 87bb48eec34f749c55350d337b5ef9710e732151
2023-06-11 19:06:29 +08:00
BUAADreamer
4adbb95b03 add some
Former-commit-id: 6982a53ed1f6f9fa03e99623b98fff56bf00317e
2023-06-11 18:55:53 +08:00
BUAADreamer
5b93ca6c39 add code for reading from multi files in one directory
Former-commit-id: 9b80cf08b9f0d4aee896b228fb76399e9a7c9d8b
2023-06-10 16:27:30 +08:00
BUAADreamer
ef6c5ae18a add code for reading from multi files in one directory
Former-commit-id: b7ebb83a96619e5111b0faa9da9d0feb8d9cdff0
2023-06-10 15:53:47 +08:00
hiyouga
03c92c79ff tiny fix
Former-commit-id: c9c795f9c7cd2228410a12af4ec10d3b59be87db
2023-06-07 16:42:31 +08:00
hiyouga
fc6091e118 tiny fix
Former-commit-id: 267703f1db20e5b39c2e80a37e028d908af7ffb1
2023-06-07 16:02:07 +08:00
hiyouga
025670b4f6 tiny fix
Former-commit-id: 4a9bc72d90b65db80b375cd141484abfbb0dcf0d
2023-06-07 12:58:14 +08:00
hiyouga
d6b32dd9ea add templates
Former-commit-id: 1d0686b2cb9edd4a7d320d11e65b50aab0ebd038
2023-06-07 12:40:44 +08:00
hiyouga
f57dae4a1a add belle template
Former-commit-id: c489c8ecbaaa511ddc7dc1de685981531eedd38c
2023-06-07 12:30:11 +08:00
hiyouga
5e2ec2d104 tiny fix
Former-commit-id: 7115bee4310888ec2e5f104e8d2c1f7127fb6ce6
2023-06-07 12:08:39 +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
12094c1db5 recover logging
Former-commit-id: d74014496e4ccda2de4482075a91747854facddd
2023-06-06 21:36:37 +08:00
hiyouga
bf5ad34196 support distributed quantized training
Former-commit-id: 74ff23a4f36f859f791f7b4be6f1877edc68f12f
2023-06-06 17:39:41 +08:00
hiyouga
ac6f50dedf add API demo from #1
Former-commit-id: c955edcef168da44257c5b50d7bc59266d909782
2023-06-05 21:32:18 +08:00
hoshi-hiyouga
8fd9ef924d Merge pull request #11 from hiyouga/api
Api

Former-commit-id: 9b2f524ea7f3a28f7413b8ce67e585f1596566a5
2023-06-05 20:58:02 +08:00
hiyouga
a409e1f42c fix bug in web demo
Former-commit-id: 01d6d7a910b9845a0ea38632661ce813e5cfe3a2
2023-06-05 17:58:29 +08:00
hiyouga
3f5869111b increase max length in cli demo
Former-commit-id: 0113cdb12728419022b5c01c932a5d52e626a200
2023-06-05 16:49:14 +08:00
hiyouga
f9c51a8340 implement stream generating
Former-commit-id: 6cc9535975d823ffef7e1686749b69b40347a8ec
2023-06-05 16:43:44 +08:00
hiyouga
a817801c0f tiny fix
Former-commit-id: 3c5da617cdab34c6cae038e3a06d0468ae4c6c86
2023-06-05 15:25:22 +08:00
hiyouga
063a83ab4e tiny fix
Former-commit-id: 5ce3e0056948aded120b63e365a892f9d8c3c840
2023-06-04 16:35:50 +08:00
hiyouga
eebe71699b tiny fix
Former-commit-id: a98ebf62fb82ffe5aaaea6a1ce3d4c60d23a5728
2023-06-04 12:55:40 +08:00
hiyouga
5f44112cf5 support QLoRA
Former-commit-id: d89597e28fe9b91246e58c55eeb9082436940481
2023-06-04 00:08:56 +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
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
ec48d06b9e remove unused code
Former-commit-id: f5c784dd75e152edb162ea02bf601d279f9de893
2023-06-03 00:10:54 +08:00
hiyouga
217b89cf7e add wechat
Former-commit-id: 0251b1788081a81773d87ca5ca2bac8428489852
2023-06-02 21:47:10 +08:00
hiyouga
382afc3822 tiny fix
Former-commit-id: d479aaf07a0997737da90c00558c42b32109928f
2023-06-02 19:02:25 +08:00
hiyouga
09997a25d3 fix layer norm name in PPO
Former-commit-id: 3ffc2efb997b717b3efad92a507584276e4bdfa1
2023-06-02 17:30:01 +08:00
hiyouga
58c8b29913 fix #1
Former-commit-id: b8e556e4632a79251a2225075c570337eeafa559
2023-06-02 14:25:00 +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
1512711ca2 fix bug at inference
Former-commit-id: df9b41af4401006b8040eb53c44dd290b604e0eb
2023-05-31 18:11:53 +08:00
hiyouga
a79df3500b update readme
Former-commit-id: 4054e85c664c541f435619baebcd687f80445d4a
2023-05-31 16:57:43 +08:00
hiyouga
693c049eac support BLOOM models
Former-commit-id: 1314b6ea39a01aa8ac325e1d875ac013d43aec45
2023-05-31 16:54:06 +08:00
hoshi-hiyouga
7492e8f208 Merge pull request #1 from mMrBun/main
Support conversation via API.

Former-commit-id: 5e64c7446718845444a14c5643c1bc5819562d60
2023-05-30 16:34:00 +08:00
hiyouga
181c776b58 remove dummy code
Former-commit-id: e6bc89d280945bbf48281107145c40a41d7cbd56
2023-05-30 16:28:00 +08:00
mMrBun
ef0aceaa50 Support conversation via API.
Former-commit-id: 57cfe9128ce1781853555b22b502ec85b8b01941
2023-05-30 15:00:28 +08:00