110 Commits

Author	SHA1	Message	Date
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
hiyouga	1d8a1878ea	fix PPO trainer	2023-08-02 19:10:23 +08:00
hiyouga	b5ba87952a	update ppo trainer	2023-08-02 18:46:41 +08:00
hiyouga	286f7be346	fix memory leak of PPO trainer	2023-08-02 17:41:34 +08:00
hiyouga	ac88ce5233	fix RM save model	2023-08-01 11:56:17 +08:00
hiyouga	973a638665	release v0.1.4	2023-08-01 10:08:47 +08:00
hiyouga	d3a0692d4d	fix inference	2023-08-01 00:06:48 +08:00
hiyouga	9cb1f119a4	fix arg check	2023-07-31 23:48:57 +08:00
hiyouga	62dca5bb82	update readme	2023-07-31 23:42:32 +08:00
hiyouga	0411a4b3e1	support streaming data, fix #284 #274 #268	2023-07-31 23:33:00 +08:00
hiyouga	00efa8a07f	fix #242	2023-07-25 17:04:02 +08:00
hiyouga	0b6150bc31	release v0.1.3	2023-07-21 16:48:34 +08:00
hiyouga	d2f18197e3	fix save function	2023-07-21 14:09:07 +08:00
hiyouga	ed0e186a13	update web UI, support rm predict #210	2023-07-21 13:27:27 +08:00
hiyouga	67a2773074	simplify code	2023-07-20 15:08:57 +08:00
hiyouga	d1d8e8bae1	tiny fix	2023-07-19 22:53:46 +08:00