61 Commits

Author	SHA1	Message	Date
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
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	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	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	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	57146c101f	fix #649	2023-08-23 20:21:15 +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
hiyouga	d125218cde	support bf16 ppo #551	2023-08-18 00:40:32 +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	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
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	b5ba87952a	update ppo trainer	2023-08-02 18:46:41 +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