94 Commits

Author	SHA1	Message	Date
hiyouga	dbaef776a1	tiny fix	2023-09-21 19:52:06 +08:00
hiyouga	338b8664ed	fix #944	2023-09-21 19:51:02 +08:00
hiyouga	ace3f85a72	tiny fix	2023-09-21 15:25:29 +08:00
statelesshz	b3e41c6d49	support export model on Ascend NPU	2023-09-20 10:26:02 +08:00
hiyouga	469f859161	add tests.cal_flops.py	2023-09-16 23:40:41 +08:00
hiyouga	0b5f970c05	fix #913	2023-09-15 20:58:28 +08:00
hiyouga	8857e45602	fix #887	2023-09-14 17:56:58 +08:00
mmbwf	30fb721f12	Update utils.py ... Fix parameters load error.	2023-09-14 15:38:04 +08:00
hiyouga	7ba57d5b14	fix ppo save model	2023-09-12 16:25:29 +08:00
hiyouga	d4be857e23	fix #762 #814	2023-09-12 16:10:10 +08:00
hiyouga	0fbece85a7	update flashattn, fix ppo save model	2023-09-11 17:25:36 +08:00
hiyouga	b218c271ed	remove PeftTrainer	2023-09-10 22:23:23 +08:00
hiyouga	d8aa1404be	support FlashAttention2	2023-09-10 20:43:56 +08:00
hiyouga	815b92e698	fix #850	2023-09-10 14:22:03 +08:00
hiyouga	a51b7c98ac	fix lora target	2023-09-09 17:04:45 +08:00
hiyouga	bca1a247bc	support lora target auto find	2023-09-09 15:38:37 +08:00
hiyouga	d8d82ca281	fix chatglm2 tokenizer	2023-09-09 13:50:29 +08:00
hiyouga	90bd085ae4	fix bug in DPO data collator	2023-09-08 20:45:07 +08:00
hiyouga	b34797a845	fix #761	2023-09-08 20:22:18 +08:00
hiyouga	8ea32e4046	change to right-padding, update reward score #803	2023-09-08 20:04:31 +08:00
hiyouga	f5351c18e1	update requirements	2023-09-07 19:26:25 +08:00
hiyouga	5a9970dbef	fix #818	2023-09-07 19:19:53 +08:00
hiyouga	ed1c2c5557	add deepspeed check in PPO training	2023-09-07 19:12:40 +08:00
hiyouga	e2bf7c3bad	fix #809	2023-09-07 19:04:32 +08:00
hiyouga	85b1f6632a	fix baichuan templates	2023-09-07 18:54:14 +08:00
hiyouga	a9d1fb72f7	refactor dataset_attr, add eos in pt, fix #757	2023-09-01 19:00:45 +08:00
codemayq	f7fdc088d4	add dataset stage check	2023-08-30 16:23:08 +08:00
hiyouga	2de1a7610a	fix import error	2023-08-23 20:45:03 +08:00
hiyouga	57146c101f	fix #649	2023-08-23 20:21:15 +08:00
hiyouga	5235b15c91	fix #617	2023-08-21 18:16:11 +08:00
hiyouga	0676497104	fix ppo trainer #551	2023-08-20 14:07:11 +08:00
hiyouga	9c9009f49f	Release v0.1.7	2023-08-18 17:21:27 +08:00
hiyouga	d75e377b0f	tiny fix	2023-08-18 13:07:35 +08:00
hiyouga	53e33418d0	support ppo score norm (trl 0.5.1.dev required)	2023-08-18 12:02:42 +08:00
hiyouga	9020524418	fix PPO trainer #551 , update readme	2023-08-18 11:43:10 +08:00
hiyouga	58f13e22da	update training resuming	2023-08-18 01:41:17 +08:00
hoshi-hiyouga	7252903245	Merge branch 'main' into main	2023-08-18 01:37:23 +08:00
hiyouga	d125218cde	support bf16 ppo #551	2023-08-18 00:40:32 +08:00
hiyouga	9f4c2adc9a	fix ChatGLM2 ppo #527 #528	2023-08-18 00:34:59 +08:00
hiyouga	be21fc83f9	fix generation bug #532	2023-08-17 22:21:34 +08:00
hiyouga	b0ed0dec5e	fix streaming in pt stage #548 #549	2023-08-17 17:59:26 +08:00
hiyouga	d9e62711a3	fix generation	2023-08-16 22:39:54 +08:00
hiyouga	7407d9daa1	fix system prompt	2023-08-16 01:35:52 +08:00
hiyouga	af6c011fcb	fix ChatGLM RLHF	2023-08-15 11:19:20 +08:00
hiyouga	80b4053602	alert pad_token source	2023-08-15 00:07:56 +08:00
hiyouga	d019956808	fix ChatGLM lm_head #494	2023-08-14 14:14:48 +08:00
hiyouga	ec94274ca1	web UI integrating RLHF	2023-08-14 10:48:47 +08:00
hiyouga	2f2fd55d81	fix #480	2023-08-14 00:23:56 +08:00
hiyouga	9dc6a296e3	tiny fix	2023-08-12 22:02:43 +08:00
hiyouga	8545c11c45	fix rope scaling	2023-08-12 22:00:01 +08:00