7 Commits

Author	SHA1	Message	Date
hiyouga	5eef8d5d98	use low_cpu_mem_usage to speed up loading ... Former-commit-id: 7891e4c200566a4a47088e93efd1fbebcb46528e	2023-06-03 18:19:01 +08:00
hiyouga	9b8b6623ac	add logits processor ... Former-commit-id: f6f4b1554ae1e8849b437d705ffa34ce7ebd56bb	2023-06-03 16:34:54 +08:00
hiyouga	58c8b29913	fix #1 ... Former-commit-id: b8e556e4632a79251a2225075c570337eeafa559	2023-06-02 14:25:00 +08:00
hiyouga	693c049eac	support BLOOM models ... Former-commit-id: 1314b6ea39a01aa8ac325e1d875ac013d43aec45	2023-05-31 16:54:06 +08:00
hiyouga	304be6dc28	fix checkpoint loading ... Former-commit-id: d31aa5c2c0bcb6a4ef4a62e21693548dd9acaae6	2023-05-29 17:43:16 +08:00
hiyouga	35d04a2c05	tiny fix ... Former-commit-id: eae79707d31fd8be2cf4bee4d610557bbd49f6e7	2023-05-29 09:42:29 +08:00
hiyouga	17024ebc1a	Initial commit ... Former-commit-id: 5ca8e1d63727e7bcb8cab16542c763c47e48184a	2023-05-28 18:09:04 +08:00