21 Commits

Author	SHA1	Message	Date
hiyouga	54c6905937	add docstrings, refactor logger	2024-09-08 00:56:56 +08:00
hiyouga	dabad5570b	update get template	2024-09-04 22:36:20 +08:00
hiyouga	47ea97fb1b	lazy image load	2024-09-04 02:27:08 +08:00
hiyouga	a244f143f4	optimize predict vram	2024-08-30 23:08:45 +08:00
moontidef	40908a36fa	fix: rename optimzer to optimizer	2024-08-07 10:05:01 +08:00
hiyouga	beec77a089	fix metrics #4786	2024-07-17 00:47:00 +08:00
hiyouga	d774b94f12	support batch_eval_metrics, fix #4826	2024-07-17 00:33:00 +08:00
hiyouga	29ebcd75d5	fix up	2024-07-15 01:04:56 +08:00
codingma	76f3bbcfc0	1. add custom eval dataset support ... 2. merge load dataset and split dataset function	2024-07-05 15:52:10 +08:00
hiyouga	821bb6660e	remove rlhf support for chatglm2&3	2024-07-02 23:03:17 +08:00
hiyouga	c13ae2df19	upcast logits	2024-07-02 22:32:05 +08:00
hiyouga	c47ab6c072	improve rlhf	2024-07-02 22:23:08 +08:00
hiyouga	6f63050e1b	add Gemma2 models	2024-06-28 01:26:50 +08:00
hiyouga	8baf3b22b0	refactor pissa, improve llamaboard	2024-06-28 01:04:24 +08:00
hiyouga	095fab58d3	tiny fix about badam	2024-06-25 01:54:53 +08:00
Jonery	5c2ff1b749	Cleaner integration.	2024-06-19 12:29:40 +08:00
Jonery	0f72aac8c9	Support distributed BAdam.	2024-06-18 12:27:47 +08:00
hiyouga	38b6b0f52e	tiny fix	2024-06-16 01:06:41 +08:00
hiyouga	d87108daa6	add license	2024-06-15 17:54:33 +08:00
hiyouga	74f96efef9	rename files	2024-06-07 00:09:06 +08:00
hiyouga	308edbc426	rename package	2024-05-16 18:39:08 +08:00