Commit Graph

612 Commits

Author SHA1 Message Date
hiyouga
62941919e8 fix chatglm template
Former-commit-id: 8aaaa132d4
2023-09-08 14:45:58 +08:00
hiyouga
f225a71445 update requirements
Former-commit-id: f5351c18e1
2023-09-07 19:26:25 +08:00
hiyouga
091326dc9f fix #818
Former-commit-id: 5a9970dbef
2023-09-07 19:19:53 +08:00
hiyouga
5030f05126 add deepspeed check in PPO training
Former-commit-id: ed1c2c5557
2023-09-07 19:12:40 +08:00
hiyouga
e6fa0229f4 fix #809
Former-commit-id: e2bf7c3bad
2023-09-07 19:04:32 +08:00
hiyouga
f74b980650 fix baichuan templates
Former-commit-id: 85b1f6632a
2023-09-07 18:54:14 +08:00
hiyouga
51f662860d update baichuan2 template
Former-commit-id: 0531886e1f
2023-09-06 21:43:06 +08:00
hiyouga
f9aee17f9d add Baichuan2 models
Former-commit-id: 62ce65c628
2023-09-06 18:36:04 +08:00
Q
a46ffa68ff fix utils.py bug
Former-commit-id: a19fc2ebf7
2023-09-05 10:38:01 +08:00
hiyouga
3b5a9c60b6 fix #763
Former-commit-id: 370bdb6e43
2023-09-01 23:13:05 +08:00
hiyouga
a4fd976048 refactor dataset_attr, add eos in pt, fix #757
Former-commit-id: a9d1fb72f7
2023-09-01 19:00:45 +08:00
codemayq
ea74e5a81b update llama2 template
Former-commit-id: 0bcc489c42
2023-08-30 16:23:56 +08:00
codemayq
c955d9267c add dataset stage check
Former-commit-id: f7fdc088d4
2023-08-30 16:23:08 +08:00
codingma
00c3cd5454 Merge pull request #651 from hiyouga/feature-dataset_stage
add dataset stage

Former-commit-id: 9b4d16d040
2023-08-28 16:03:45 +08:00
codemayq
292abfd438 add text format dataset preview in webui
Former-commit-id: 24e68d29f2
2023-08-24 19:45:36 +08:00
codemayq
2b979d39f2 add stage in DatasetAttr
Former-commit-id: ba94c8729d
2023-08-23 20:54:53 +08:00
hiyouga
0d0232479f fix import error
Former-commit-id: 2de1a7610a
2023-08-23 20:45:03 +08:00
hiyouga
38080233a5 fix #649
Former-commit-id: 57146c101f
2023-08-23 20:21:15 +08:00
codemayq
4b29d9d2b0 add dataset stage and filter dataset when stage chosen in webui
Former-commit-id: c0e4d1e81b
2023-08-23 18:54:23 +08:00
hiyouga
2accde2cb7 fix webui
Former-commit-id: 1c702ad538
2023-08-23 11:03:35 +08:00
codemayq
433a96b8c2 fix quantization bit is ""
Former-commit-id: a7cc6c4140
2023-08-23 10:08:17 +08:00
codemayq
7b82ea708d fix quantization is ""
Former-commit-id: ec2047b064
2023-08-23 10:04:03 +08:00
hiyouga
802494e20a update template
Former-commit-id: 4318347d3f
2023-08-22 19:46:09 +08:00
hiyouga
d05d535a58 fix #617
Former-commit-id: 5235b15c91
2023-08-21 18:16:11 +08:00
hiyouga
e6f4eab4ab fix #608
Former-commit-id: 02d69b6fde
2023-08-21 17:49:36 +08:00
hiyouga
d3bef03dc6 fix baichuan template for training #597 #616
Former-commit-id: 0a3f698425
2023-08-21 17:41:51 +08:00
hiyouga
fa15416024 fix #595
Former-commit-id: 5c052836a0
2023-08-20 16:40:00 +08:00
beat4ocean
c81d1d40fa fix KeyError: 'lang' bug
Former-commit-id: 7b45de6b9f
2023-08-20 15:32:36 +08:00
hiyouga
570ccc3618 fix ppo trainer #551
Former-commit-id: 0676497104
2023-08-20 14:07:11 +08:00
hiyouga
acaac6df9e Release v0.1.7
Former-commit-id: 9c9009f49f
2023-08-18 17:21:27 +08:00
hiyouga
9f1688924d tiny fix
Former-commit-id: d75e377b0f
2023-08-18 13:07:35 +08:00
hiyouga
b88f0b396c support ppo score norm (trl 0.5.1.dev required)
Former-commit-id: 53e33418d0
2023-08-18 12:02:42 +08:00
hiyouga
03edfd07e7 fix PPO trainer #551 , update readme
Former-commit-id: 9020524418
2023-08-18 11:43:10 +08:00
hiyouga
fceca0bb6a update training resuming
Former-commit-id: 58f13e22da
2023-08-18 01:41:17 +08:00
hoshi-hiyouga
49d4ae3704 Merge branch 'main' into main
Former-commit-id: 7252903245
2023-08-18 01:37:23 +08:00
hiyouga
66771352bb support bf16 ppo #551
Former-commit-id: d125218cde
2023-08-18 00:40:32 +08:00
hiyouga
caf4a61e21 fix ChatGLM2 ppo #527 #528
Former-commit-id: 9f4c2adc9a
2023-08-18 00:34:59 +08:00
hiyouga
623a34b16f fix generation bug #532
Former-commit-id: be21fc83f9
2023-08-17 22:21:34 +08:00
hiyouga
a46f277477 fix streaming in pt stage #548 #549
Former-commit-id: b0ed0dec5e
2023-08-17 17:59:26 +08:00
hiyouga
3021a01b71 fix baichuan and intern template
Former-commit-id: 892fd39373
2023-08-17 01:27:20 +08:00
hiyouga
048f99354f fix generation
Former-commit-id: d9e62711a3
2023-08-16 22:39:54 +08:00
hiyouga
edc15c62fa fix system prompt
Former-commit-id: 7407d9daa1
2023-08-16 01:35:52 +08:00
hiyouga
2ceaecfb42 fix baichuan template #481
Former-commit-id: 273135f595
2023-08-15 11:38:21 +08:00
hiyouga
a9ab8f71d7 fix ChatGLM RLHF
Former-commit-id: af6c011fcb
2023-08-15 11:19:20 +08:00
hiyouga
d15fe288df alert pad_token source
Former-commit-id: 80b4053602
2023-08-15 00:07:56 +08:00
hiyouga
02a61b08b1 update webui
Former-commit-id: 9d0f6214b6
2023-08-14 22:45:26 +08:00
hoshi-hiyouga
d021d31a9c Merge pull request #511 from hiyouga/feature-autoTemplate
add template match and stage in webui

Former-commit-id: adb0f186e9
2023-08-14 22:44:04 +08:00
codemayq
4a4623cf2d auto match template when change model_name
Former-commit-id: 0bf892ff1a
2023-08-14 20:56:05 +08:00
codemayq
ee7da14f81 add template match and stage in webui
Former-commit-id: 79c68e5527
2023-08-14 20:42:59 +08:00
hiyouga
2aa2c363ad fix ChatGLM lm_head #494
Former-commit-id: d019956808
2023-08-14 14:14:48 +08:00