5 Commits

Author	SHA1	Message	Date
saibit	46ad3be059	update deepspeed wrapper	2025-04-24 11:26:36 +08:00
sharlynxy	abf2c44bc5	Dynamically set device in deepspeed wrapper (#2) ... * get device type from model * add logger warning * format * format * format	2025-04-23 18:57:19 +08:00
Robert	c8af252a44	refactor	2025-04-22 16:19:14 +08:00
saibit	7c61c0dfe0	Add autocast warpper for forward functions in deepspeed_utils.py to try aligning precision when using mixed precision in training process	2025-04-22 16:06:55 +08:00
Kohya S	e3ccf8fbf7	make deepspeed_utils	2024-02-27 21:30:46 +09:00