75 Commits

Author	SHA1	Message	Date
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
hiyouga	1836c020c5	update readme	2023-08-12 21:00:11 +08:00
hiyouga	fa940c17b8	support rope scaling, fix #475 #476 #478	2023-08-12 20:46:27 +08:00
hiyouga	dd51c24203	fix unusual output of 8bit models #278 #391	2023-08-12 00:25:29 +08:00
hiyouga	a48cb0d474	Release v0.1.6	2023-08-11 23:25:57 +08:00
hiyouga	3ec4351cfd	support DPO training (2305.18290)	2023-08-11 03:02:53 +08:00
hiyouga	d86ea314a1	support val set in streaming mode	2023-08-09 23:00:26 +08:00
niuba	2ec68d3398	add last_checkpoint support	2023-08-09 16:39:27 +08:00
hiyouga	df946e6949	fix sft trainer	2023-08-09 16:35:03 +08:00
hiyouga	eecc4b2131	fix tokenizer #417	2023-08-08 23:59:41 +08:00
hiyouga	4b841a6b35	fix bug	2023-08-08 17:55:55 +08:00
hiyouga	a9980617f5	fix chatml template #408	2023-08-08 17:44:39 +08:00
hiyouga	081345baca	fix #376	2023-08-07 13:58:59 +08:00
hiyouga	220175ab24	update trainer	2023-08-07 13:34:35 +08:00
hiyouga	e21ae01356	fix qwen eos token	2023-08-06 13:31:17 +08:00
hiyouga	a0173c427d	fix mtloader	2023-08-03 19:29:02 +08:00
hiyouga	2780792754	fix qwen inference	2023-08-03 16:31:55 +08:00
hiyouga	87f8f830e2	support Qwen-7B, fix InternLM-7B inference	2023-08-03 15:53:32 +08:00
hiyouga	e23a3a366c	fix webui	2023-08-03 12:43:12 +08:00
hiyouga	08f180e788	modify code structure	2023-08-02 23:17:36 +08:00