113 Commits

Author	SHA1	Message	Date
hiyouga	6f7b6ae0c3	remove gc warnings in DPO&KTO ... Former-commit-id: f9a206509e	2024-06-03 22:53:54 +08:00
hoshi-hiyouga	5d96cf146e	Update trainer.py ... Former-commit-id: 24499f40dc	2024-06-03 22:08:38 +08:00
enji.zhou	e58aca0602	fix KTO Trainer Sampler ... Former-commit-id: 34a2c5087a	2024-06-03 21:32:38 +08:00
Uminosachi	0de4e1e9e2	Set scheduler_specific_kwargs to get_scheduler ... Former-commit-id: 14e97dc119	2024-05-31 13:45:39 +09:00
hiyouga	468d0e7ed1	10x generate in ppo w/ zero3 ... https://github.com/huggingface/trl/pull/1483 Former-commit-id: 65cd8bdbdb	2024-05-29 00:23:23 +08:00
hiyouga	bfac965f9c	update dpo, kto trainer ... Former-commit-id: 7c8e01bb74	2024-05-29 00:14:29 +08:00
hiyouga	14f6cc2b7c	clean kto trainer ... Former-commit-id: 900e1ea622	2024-05-28 21:43:26 +08:00
hiyouga	4807c11db8	support SimPO #3900 ... Former-commit-id: cb63b32986	2024-05-26 23:46:33 +08:00
hiyouga	3e729798df	refactor data preprocessing, fix mllm rlhf ... Former-commit-id: 3a023bca2a	2024-05-24 04:08:25 +08:00
hiyouga	519d2511ae	improve data process logger ... Former-commit-id: a851056229	2024-05-18 22:02:42 +08:00
hiyouga	13d7b48efe	improve KTO impl., replace datasets ... Former-commit-id: c450ee87a3	2024-05-18 03:44:56 +08:00
enji.zhou	03956053b8	add kto ... Former-commit-id: db1d5a4f51	2024-05-17 13:09:17 +08:00
hiyouga	cae823ddf0	rename package ... Former-commit-id: 308edbc426	2024-05-16 18:39:08 +08:00