146 Commits

Author SHA1 Message Date
hiyouga
64eead3fb1 ppo support rm server
Former-commit-id: 747db4017291b0eb91946f57011bb31659056037
2023-12-03 21:38:51 +08:00
hiyouga
2279b1948e fix #1707 #1710
Former-commit-id: 03d05991f81b51826d3a4d9da214504e19a301bd
2023-12-03 11:33:12 +08:00
hiyouga
f8376b228a add xuanyuan models
Former-commit-id: 6e7af11b989e4cf97ffacbab4736e3434ff6c925
2023-12-02 00:35:29 +08:00
hiyouga
4cc08e00c7 fix gptq training
Former-commit-id: f57445c7a01b726160070c4ea29486702a650ed6
2023-12-02 00:27:15 +08:00
hiyouga
a388af4adc fix #1703
Former-commit-id: 662d9a3a4ec5ee408f5c9574c502e0af662fd821
2023-12-01 22:55:41 +08:00
hiyouga
c60e79c12e patch modelscope
Former-commit-id: bd42c229b01a0bf3ceadb8cee5ad49a060cc2d13
2023-12-01 22:53:15 +08:00
hoshi-hiyouga
9a26819a58 Merge branch 'main' into feat/support_ms
Former-commit-id: 00f5c9ee1608b98ab8f40bcafdc3edc71833257f
2023-12-01 20:23:46 +08:00
yuze.zyz
fcd61657ee remove useless code
Former-commit-id: 5a2392f105704810e9ce96c13fcc8a555726f9b8
2023-12-01 17:28:23 +08:00
tastelikefeet
eb835b693d fix bug
Former-commit-id: d9e52957e272e8133f1b37cf20d193084425e09e
2023-12-01 17:27:00 +08:00
yuze.zyz
b2200409f5 add readme
Former-commit-id: 5aa6751e52b5c2e06727c50e60218226b146b7bf
2023-12-01 16:11:30 +08:00
hiyouga
a44ba7a2b8 tiny fix
Former-commit-id: e597d3c084c8700e247bad6e26d2ee40fc3c316b
2023-12-01 15:58:50 +08:00
hoshi-hiyouga
752b3dd58d Merge pull request #1690 from billvsme/main
Improve get_current_device

Former-commit-id: d043a4e7ba2457679eeca8de5820a4f07b7d8401
2023-12-01 15:44:35 +08:00
hiyouga
9a6b694e12 fix #1696
Former-commit-id: bf6f6aeefe65b4949633648b8711525c0029c001
2023-12-01 15:34:50 +08:00
tastelikefeet
63e12226a0 add model
Former-commit-id: 8ce4d11e38518b0b4657c7e64394d471cbb0bd6d
2023-12-01 15:06:17 +08:00
billvsme
e400f2e8ad improve get_current_device
Former-commit-id: 40dfcbc3d4571ce022b6aa39db581c8b88a75b8d
2023-11-30 22:40:35 +08:00
hiyouga
1c43fb6a41 add models
Former-commit-id: 509abe8864ada29ac7fa0f636b662531c8dd3a33
2023-11-30 19:16:13 +08:00
yuze.zyz
45925e4a9c fix
Former-commit-id: fb2204c183ae8c061ed6ec7f4f1bfbb0b4900c9b
2023-11-29 21:43:58 +08:00
yuze.zyz
e08e0e5814 support ms
Former-commit-id: d38a2e7341100902b6c761895b1fe6191c905d06
2023-11-29 20:36:55 +08:00
hiyouga
ae1048db6d fix #1659
Former-commit-id: 475a3fa0f4c09d4cfd55ec66271a6d3c9eb5f4d2
2023-11-28 20:52:28 +08:00
hiyouga
5f2943dc84 support Yi-34B-Chat models
Former-commit-id: ff1c289229ee382d3e76578bbb6a5e299b969ded
2023-11-23 19:31:49 +08:00
hiyouga
f06c4c8f7a update ppo trainer
Former-commit-id: 5021062493ed63ad1f6133cfb543e4e7f528d2cc
2023-11-20 21:39:15 +08:00
hiyouga
48d6d925f7 fix #1558
Former-commit-id: 1740131d63d32aefc0370441baf4716ddb5ebcfe
2023-11-19 14:15:47 +08:00
hiyouga
d3c4881ccb fix packages
Former-commit-id: 999bc0ed93d15b5d1082c8f706a7d17c95933d93
2023-11-17 16:11:48 +08:00
Shaowen Wang
4ea3144554 Fix: Change rouge-chinese package name to rouge_chinese
To reproduce:
python:
importlib.util.find_spec('rouge-chinese') -> None
importlib.util.find_spec('rouge_chinese') -> ModuleSpec(name='rouge_chinese'...)
from rouge_chinese import Rouge
print(Rouge.__module__) -> rouge_chinese
Former-commit-id: 397e9489849d80517b0c467852a7092a0a9626f2
2023-11-16 20:12:35 -06:00
hiyouga
3f53155a90 fix bug in web ui
Former-commit-id: 6efa38be46ed536f80fc67002f23862edcb9df8d
2023-11-16 15:21:24 +08:00
hiyouga
e4f97615f0 update ppo and demo in webui
Former-commit-id: 7537dd434f4c0f0bde06bd8c2ac69bf622772316
2023-11-16 14:55:26 +08:00
hiyouga
627212e48b tiny fix
Former-commit-id: 83cee2a6049b8287de1b5ebf41b2a0728e235b11
2023-11-16 03:27:19 +08:00
hiyouga
678052a7ef fix rlhf callback
Former-commit-id: 1817ffc86fe3463ea91e9359c0e3611979a9d53e
2023-11-16 03:26:19 +08:00
hiyouga
f441932bd1 support full-parameter PPO
Former-commit-id: ce783036001397a20b0b4c5da2fea6d0c03389d2
2023-11-16 02:08:04 +08:00
hiyouga
3e0b76650a update readme and constants
Former-commit-id: 1e19cf242a1f843b590feefbe24b2cc0a17712b5
2023-11-15 18:04:37 +08:00
hiyouga
06a4820836 disentangle model from tuner and rename modules
Former-commit-id: 4736344eb1595ee023a50d49e8118f4eee46305f
2023-11-15 16:29:09 +08:00
hiyouga
4a767e5593 release v0.2.2, fix #1478 #1466
Former-commit-id: 35cc1e28f675889c44f75a0a3194005c7f23631b
2023-11-13 23:09:05 +08:00
hiyouga
125587b187 refactor evaluation, upgrade trl to 074
Former-commit-id: 442aefb925c4ff02b98aa30c49c2e01d04f6496a
2023-11-13 22:20:35 +08:00
hiyouga
982e0e79c2 fix flashattn warning
Former-commit-id: 4bd8e3906d09bf6ec4b8f6b553a347fca9db4f80
2023-11-10 18:34:54 +08:00
hiyouga
0fbaa42752 refactor constants
Former-commit-id: 3697a3dc9a0be8141951dfe65812844f66059517
2023-11-10 14:16:10 +08:00
hiyouga
38755bced7 add template, modify datasets
Former-commit-id: 386f590209e466b51c17a7ac8cee55fc3ce928d7
2023-11-09 15:53:23 +08:00
hiyouga
1f2c56bff9 delete file
Former-commit-id: 479d0af2dc4ab8282b9d55aba1b03ab3a54f400b
2023-11-07 16:20:12 +08:00
hiyouga
3d40bdb600 upgrade peft, fix #1088 #1411
Former-commit-id: b2a60905f384ada92618bf21301fe96dac1c10bf
2023-11-07 16:13:36 +08:00
hiyouga
a919b6a478 update templates
Former-commit-id: a7eeb8e17c2f23f16732f5a5d767b39bcc1ac517
2023-11-06 12:25:47 +08:00
hiyouga
034b658348 fix deepseek template
Former-commit-id: d08f5e8a147f1929567d42b6bed8bc998c2a866d
2023-11-05 13:08:46 +08:00
hiyouga
04107b7af6 support deepseek coder #1378
Former-commit-id: 2a8a25819524e84a5e6e907923c47693f8b7a48d
2023-11-05 12:51:03 +08:00
hiyouga
6493f6d2e9 fix #1316
Former-commit-id: f4e4a04529a60b4c4ccc66cbf67f6e951fbc68d3
2023-10-31 11:32:08 +08:00
hiyouga
d48478ef88 update constants
Former-commit-id: f28a034a9b74630a56314446bd0f103c086bda60
2023-10-29 13:30:20 +08:00
hiyouga
bf0faf129d fix vicuna template
Former-commit-id: 52fc24d1664bc701f43e2bff8b3faded795b929c
2023-10-27 22:15:25 +08:00
hiyouga
5705c82cd8 fix chatglm3 template
Former-commit-id: 4117f388279ca43eb46def195c21e7051aefd0c7
2023-10-27 21:12:06 +08:00
hiyouga
8a76b1e499 support chatglm3
Former-commit-id: 1c0ab9a908dedf0ad69ad5741a23465da02006d9
2023-10-27 19:16:28 +08:00
hiyouga
d18c708f14 fix openchat template
Former-commit-id: 8fdff07e1f056afa5fe39fe794c6af030fc5f225
2023-10-21 01:25:42 +08:00
hiyouga
95697652f1 fix #1232
Former-commit-id: b665e9e133bf2f6f10346c374eb0de8a96dd5c7e
2023-10-20 23:28:52 +08:00
hiyouga
0503d45782 fix eval resuming in webui
Former-commit-id: 273745f9b9d117d4053afc1746108af95b0a51a4
2023-10-15 15:45:38 +08:00
hiyouga
99592478c9 tiny fix
Former-commit-id: 3ad8c92ecabaf2c169e53a8485687b4d04a772e7
2023-10-15 05:02:48 +08:00