1963 Commits

Author SHA1 Message Date
hiyouga
4410387859 Release v0.1.8
Former-commit-id: ccb3553576164113c31be714a0295ea82321d67d
2023-09-11 17:31:34 +08:00
hiyouga
33bab0e7c1 update flashattn, fix ppo save model
Former-commit-id: 0fbece85a70222e5262a2295203de07ffe648fda
2023-09-11 17:25:36 +08:00
hiyouga
6a71361a54 remove PeftTrainer
Former-commit-id: b218c271edfb07006ddc34b1aca404088de6c528
2023-09-10 22:23:23 +08:00
hiyouga
8ab5566dc0 support FlashAttention2
Former-commit-id: d8aa1404bee9842f3e4cd037ad8d66c85470ac37
2023-09-10 20:43:56 +08:00
hiyouga
85aa16f6c6 fix #850
Former-commit-id: 815b92e698562bfae6eb9a6fa1b612a05d43ed67
2023-09-10 14:22:03 +08:00
hiyouga
f865d0bd51 fix lora target
Former-commit-id: a51b7c98acc599de5ed2eaeeebe7b184105722c5
2023-09-09 17:04:45 +08:00
hiyouga
c818a7ff60 support lora target auto find
Former-commit-id: bca1a247bcef51dced59655c8a14c197569367ca
2023-09-09 15:38:37 +08:00
hiyouga
c6265e6969 fix chatglm2 tokenizer
Former-commit-id: d8d82ca281811c20c89cc03dd00f69735515d6cf
2023-09-09 13:50:29 +08:00
hiyouga
43a20c67d4 fix bug in DPO data collator
Former-commit-id: 90bd085ae4c2775f1e82e045ab0157a451774082
2023-09-08 20:45:07 +08:00
hiyouga
405df0f63d fix #761
Former-commit-id: b34797a845dec8f6daea59e3c353b8a8f8830100
2023-09-08 20:22:18 +08:00
hiyouga
9ed4bb63d4 change to right-padding, update reward score #803
Former-commit-id: 8ea32e4046d75ddfa9517669e9de9f48fea720c6
2023-09-08 20:04:31 +08:00
hiyouga
62941919e8 fix chatglm template
Former-commit-id: 8aaaa132d49b4c758256a6159270cea4351f946c
2023-09-08 14:45:58 +08:00
hiyouga
f225a71445 update requirements
Former-commit-id: f5351c18e15cd26fe628ed364eaa4ecd49874596
2023-09-07 19:26:25 +08:00
hiyouga
091326dc9f fix #818
Former-commit-id: 5a9970dbef3a6975ce5ec6ac2bef19182c75b662
2023-09-07 19:19:53 +08:00
hiyouga
5030f05126 add deepspeed check in PPO training
Former-commit-id: ed1c2c5557bb2714c3341294f0ea86f6496d4b0c
2023-09-07 19:12:40 +08:00
hiyouga
e6fa0229f4 fix #809
Former-commit-id: e2bf7c3badbd5d2fd513ca7a00bd74d9c0d62d07
2023-09-07 19:04:32 +08:00
hiyouga
f74b980650 fix baichuan templates
Former-commit-id: 85b1f6632a752029dabdaed87c58986deb3a6b1d
2023-09-07 18:54:14 +08:00
hiyouga
51f662860d update baichuan2 template
Former-commit-id: 0531886e1f534217dc3c9c0775d29fcf77ff7f5f
2023-09-06 21:43:06 +08:00
hiyouga
f9aee17f9d add Baichuan2 models
Former-commit-id: 62ce65c6282d2bbcb765354acc2819cc3e983a46
2023-09-06 18:36:04 +08:00
Q
a46ffa68ff fix utils.py bug
Former-commit-id: a19fc2ebf7a77f95caae360bf4d61533a1d7cc74
2023-09-05 10:38:01 +08:00
hiyouga
3b5a9c60b6 fix #763
Former-commit-id: 370bdb6e4309db03e26cad311fb13e5cbb1fc1bf
2023-09-01 23:13:05 +08:00
hiyouga
a4fd976048 refactor dataset_attr, add eos in pt, fix #757
Former-commit-id: a9d1fb72f791ae57a4d12f4e3a7e2abccf6a7077
2023-09-01 19:00:45 +08:00
codemayq
ea74e5a81b update llama2 template
Former-commit-id: 0bcc489c42fc41b2c7ee51cced2dc995256a02d8
2023-08-30 16:23:56 +08:00
codemayq
c955d9267c add dataset stage check
Former-commit-id: f7fdc088d49564f7d436fd445e7e1987a9a00a0b
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: 9b4d16d040a4d2634951483958475a09148011e2
2023-08-28 16:03:45 +08:00
codemayq
292abfd438 add text format dataset preview in webui
Former-commit-id: 24e68d29f2c411ef4ff4d147e45cda664f332b65
2023-08-24 19:45:36 +08:00
codemayq
2b979d39f2 add stage in DatasetAttr
Former-commit-id: ba94c8729dd8c90bedf7079a6978e150fc92b737
2023-08-23 20:54:53 +08:00
hiyouga
0d0232479f fix import error
Former-commit-id: 2de1a7610a78e41680970b9f308741f98df489fa
2023-08-23 20:45:03 +08:00
hiyouga
38080233a5 fix #649
Former-commit-id: 57146c101f3e8f688b016a44c85e8ad5d1b6f938
2023-08-23 20:21:15 +08:00
codemayq
4b29d9d2b0 add dataset stage and filter dataset when stage chosen in webui
Former-commit-id: c0e4d1e81b41c9a36291d8bee46d7d807c898c21
2023-08-23 18:54:23 +08:00
hiyouga
2accde2cb7 fix webui
Former-commit-id: 1c702ad5385bec68d50a6d4da83b3fa10f1dc773
2023-08-23 11:03:35 +08:00
codemayq
433a96b8c2 fix quantization bit is ""
Former-commit-id: a7cc6c4140c23f3b41985a481af69964b87e0feb
2023-08-23 10:08:17 +08:00
codemayq
7b82ea708d fix quantization is ""
Former-commit-id: ec2047b064bdb6a6e084aa268ee3de454f33dd59
2023-08-23 10:04:03 +08:00
hiyouga
802494e20a update template
Former-commit-id: 4318347d3f1982c773dad1074636ec7b550770fd
2023-08-22 19:46:09 +08:00
hiyouga
d05d535a58 fix #617
Former-commit-id: 5235b15c9181f2b68f7d6caa9a6324b8570d3d0c
2023-08-21 18:16:11 +08:00
hiyouga
e6f4eab4ab fix #608
Former-commit-id: 02d69b6fdefa6b303b84fb8195a159006fe3f50a
2023-08-21 17:49:36 +08:00
hiyouga
d3bef03dc6 fix baichuan template for training #597 #616
Former-commit-id: 0a3f6984259526775b0efdb8a1b0b24f564a7239
2023-08-21 17:41:51 +08:00
hiyouga
fa15416024 fix #595
Former-commit-id: 5c052836a0a19afd556418f1f6da8db8fc3cd37d
2023-08-20 16:40:00 +08:00
beat4ocean
c81d1d40fa fix KeyError: 'lang' bug
Former-commit-id: 7b45de6b9f5a0159335614e78a106b595d726fc9
2023-08-20 15:32:36 +08:00
hiyouga
570ccc3618 fix ppo trainer #551
Former-commit-id: 0676497104eccc8a737d27890eabf1ca8713c235
2023-08-20 14:07:11 +08:00
hiyouga
acaac6df9e Release v0.1.7
Former-commit-id: 9c9009f49fdff83a29b83d8f97eb5c99e2574256
2023-08-18 17:21:27 +08:00
hiyouga
9f1688924d tiny fix
Former-commit-id: d75e377b0f6f3fd7c034676b81ddef3aab1d6901
2023-08-18 13:07:35 +08:00
hiyouga
b88f0b396c support ppo score norm (trl 0.5.1.dev required)
Former-commit-id: 53e33418d02ee0f34c783e30ae510b811308c598
2023-08-18 12:02:42 +08:00
hiyouga
03edfd07e7 fix PPO trainer #551 , update readme
Former-commit-id: 90205244186df558cd6b0000728d638348db3a10
2023-08-18 11:43:10 +08:00
hiyouga
fceca0bb6a update training resuming
Former-commit-id: 58f13e22da18babed0d2d4348474e07745da8fa5
2023-08-18 01:41:17 +08:00
hoshi-hiyouga
49d4ae3704 Merge branch 'main' into main
Former-commit-id: 725290324562e093565fae79a05341ebf64486d5
2023-08-18 01:37:23 +08:00
hiyouga
66771352bb support bf16 ppo #551
Former-commit-id: d125218cde893c7c8527ab27b4d2dfb2474c384d
2023-08-18 00:40:32 +08:00
hiyouga
caf4a61e21 fix ChatGLM2 ppo #527 #528
Former-commit-id: 9f4c2adc9a9ca8e458d3868805e077182e0d336a
2023-08-18 00:34:59 +08:00
hiyouga
623a34b16f fix generation bug #532
Former-commit-id: be21fc83f9aed0af1e5a2f83f5d5eeb36f1d283c
2023-08-17 22:21:34 +08:00
hiyouga
a46f277477 fix streaming in pt stage #548 #549
Former-commit-id: b0ed0dec5e6788a0344c09a6cc58d1116265fd68
2023-08-17 17:59:26 +08:00