766 Commits

Author SHA1 Message Date
hiyouga
f9df6c17ed fix #1550
Former-commit-id: 1bbc1be95eedf0796c0b311568dff8c75f87dfbb
2023-11-17 17:23:13 +08:00
Yuchen Han
a419122179 Update workflow.py
Former-commit-id: eeb5249d0b6ce0816e1fa47afc3a853c7b267cbf
2023-11-17 00:16:27 -08:00
Yuchen Han
ec910a87c0 Update finetuning_args.py
Former-commit-id: b24635d22b3084ad29217ef55c1dd1fa4f85a1fb
2023-11-17 00:15:51 -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
5de45bf989 fix chatglm template
Former-commit-id: ed9f7705efbed0accf4dc5c9dfa9e3e7e15e1174
2023-11-16 22:54:15 +08:00
hiyouga
8454e02313 fix web ui demo
Former-commit-id: 10ce87e088ae5934c4c3e61e9d0556cd5378308c
2023-11-16 18:41:55 +08:00
hiyouga
be0fb659d2 fix web ui demo
Former-commit-id: 1c80e9a09ed9fce0809e2b893d946306df0f9f7c
2023-11-16 17:12:23 +08:00
hiyouga
11af6c1e39 release v0.3.0
Former-commit-id: c4facc03af20d15d5b09ec77dc3742138db68f9d
2023-11-16 16:00:11 +08:00
hiyouga
11de514cc6 fix css
Former-commit-id: 08f3c114292676699a1921d4395f268a54763428
2023-11-16 15:45:38 +08: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
0ed0b8f9c5 fix bug in freeze tuning
Former-commit-id: ff52b1779c909819d0aef83d3f7ea663199cbe54
2023-11-16 14:25:11 +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
b71da932eb fix bug in PPO training
Former-commit-id: 856522a3df4bb9ddfaaa137119eceb9574873950
2023-11-16 02:32:54 +08:00
hiyouga
eb5a852dd5 fix import bug
Former-commit-id: 35b91ea34caade45dd51813b94da5177b852aa4c
2023-11-16 02:27:03 +08:00
hiyouga
f441932bd1 support full-parameter PPO
Former-commit-id: ce783036001397a20b0b4c5da2fea6d0c03389d2
2023-11-16 02:08:04 +08:00
hiyouga
0c1fab84f1 add demo mode for web UI
Former-commit-id: 8350bcf85d5e59b63da46b540c6ad860e8419d9e
2023-11-15 23:51:26 +08:00
hiyouga
3e0b76650a update readme and constants
Former-commit-id: 1e19cf242a1f843b590feefbe24b2cc0a17712b5
2023-11-15 18:04:37 +08:00
hiyouga
e30290444a support multiple modules in freeze training #1514
Former-commit-id: 4907452d955367ebe987e6deae4fd4213628f2b2
2023-11-15 17:08:18 +08:00
hiyouga
4a0be64ae6 fix imports
Former-commit-id: bbbce1f516840f722247edd37057d16502ea0557
2023-11-15 16:47:45 +08:00
hiyouga
06a4820836 disentangle model from tuner and rename modules
Former-commit-id: 4736344eb1595ee023a50d49e8118f4eee46305f
2023-11-15 16:29:09 +08:00
hiyouga
8ee48a9c9e fix #1507
Former-commit-id: 2f02f688e15db8bf8f0dad3b7284863998c207a3
2023-11-15 16:22:32 +08:00
hiyouga
fffb8ea764 add cal_lr.py
Former-commit-id: 42c8fc4fb970775159a68a123d5c7bedb701c8cf
2023-11-14 20:58:37 +08:00
hiyouga
8387f3011c fix #1494
Former-commit-id: d125ef55358837d4d76943739afeb6c70a901cd7
2023-11-14 18:07:20 +08:00
hiyouga
9176b55fe6 fix #1489
Former-commit-id: 3743b7420b823c5a56aef6100121d89b677ffb0e
2023-11-14 15:27:05 +08:00
hiyouga
5c4ddebde5 support eval remote dataset
Former-commit-id: 2d42be32c1b32b26548ea5af5fc3c810f4d668c1
2023-11-14 02:42:30 +08:00
hiyouga
4a767e5593 release v0.2.2, fix #1478 #1466
Former-commit-id: 35cc1e28f675889c44f75a0a3194005c7f23631b
2023-11-13 23:09:05 +08:00
hiyouga
37db26800c fix #424
Former-commit-id: 87390ae3b70f654d520b9aadb335c9650130a42c
2023-11-13 22:42:23 +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
55e097aaac add todo
Former-commit-id: a0c31c68c4909637b86c90c319c321fd887c4910
2023-11-10 14:38:18 +08:00
hiyouga
0fbaa42752 refactor constants
Former-commit-id: 3697a3dc9a0be8141951dfe65812844f66059517
2023-11-10 14:16:10 +08:00
hiyouga
6ee32cf71c tiny fix
Former-commit-id: 415bca900e5cc3afaddd5b06d35f472d9ead3263
2023-11-09 17:20:49 +08:00
Yanqing
fc05fd52cf Update finetuning_args.py
更新 chatglm/falcon/bloom 的 lora_target 的名称

Former-commit-id: 3684dffa14ca0551d51027467c0134b884ed1c59
2023-11-09 17:04:40 +08:00
hiyouga
4dbb52750f fix #1452
Former-commit-id: 0e86527d7fae9c9fe0df89d6fbd89035c9d83fe3
2023-11-09 16:41:32 +08:00
hiyouga
c5b202d5c6 release v0.2.1
Former-commit-id: 1db59832fd4239e8f70cff2cab8376589550a9df
2023-11-09 15:54:16 +08:00
hiyouga
38755bced7 add template, modify datasets
Former-commit-id: 386f590209e466b51c17a7ac8cee55fc3ce928d7
2023-11-09 15:53:23 +08:00
hoshi-hiyouga
28a9176784 Merge pull request #1436 from lvzii/main
fix tokenizer config changed after pretrain

Former-commit-id: 7ca32d8e69d5a2790b8ac323f6e6d0aea42600e7
2023-11-09 14:30:50 +08:00
hiyouga
f9ebc718d0 support parquet format #1446
Former-commit-id: 3df90b988bd987daa4e3a991c32ae53446481dcd
2023-11-09 14:17:40 +08:00
hiyouga
b9f42172dd fix #1438 #1439
Former-commit-id: 33422e1feff706bb7cfb235b9e40bd2e124222be
2023-11-09 13:45:10 +08:00
lvzi
13eb365eb7 fix tokenizer config changed after pretrain
Changing tokenizer's attribute at preprocessing stage will result in saving a wrong tokenizer.
for example, baichuan2

Former-commit-id: 043c316ac8913e10b2274867033f194ea92bfcd6
2023-11-08 15:50:46 +08:00
hiyouga
91f406cc99 fix ppo train and dpo eval
Former-commit-id: 01260d975477ebb8570933a1bd7f547b4dba607f
2023-11-07 22:48:51 +08:00
hiyouga
100dc4c458 fix #1422
Former-commit-id: 11c1e1e1570d3712109dd4dce831674a98841bd5
2023-11-07 19:42:01 +08:00
hiyouga
ed584b9f52 fix reward model loading
Former-commit-id: c52336d14435bc3bd98b6070cc1309b5e7d706c4
2023-11-07 17:20:51 +08:00
hiyouga
b446582bfd fix args
Former-commit-id: d92f112951a8d8b28b180c3e2f504a094a9885dd
2023-11-07 16:36:06 +08:00
hiyouga
53fcc531b5 update info
Former-commit-id: 17c64a05796cac70fc76ed728705cd60efa41cae
2023-11-07 16:28:21 +08:00
hiyouga
1f2c56bff9 delete file
Former-commit-id: 479d0af2dc4ab8282b9d55aba1b03ab3a54f400b
2023-11-07 16:20:12 +08:00
hiyouga
d843efc413 fix #1418
Former-commit-id: 7ebd63a609d4bc4f8a645c5c60be77842ebac825
2023-11-07 16:17:22 +08:00