5 Commits

Author	SHA1	Message	Date
OleehyO	fdb9820949	feat: support DeepSpeed ZeRO-3 and optimize peak memory usage ... - Add DeepSpeed ZeRO-3 configuration support - Optimize memory usage during training - Rename training scripts to reflect ZeRO usage - Update related configuration files and trainers	2025-01-12 05:33:56 +00:00
OleehyO	36427274d6	style: format import statements across finetune module	2025-01-07 05:54:52 +00:00
zR	1789f07256	format and check fp16 for cogvideox2b	2025-01-07 13:16:18 +08:00
OleehyO	93b906b3fb	docs: clarify train_frames includes padding frame ... Add docstring to train_frames field in State schema to explicitly indicate that it includes one image padding frame	2025-01-04 06:20:25 +00:00
OleehyO	91d79fd9a4	feat: add schemas module for configuration and state management ... Add Pydantic models to handle: - CLI arguments and configuration (Args) - Model components and pipeline (Components) - Training state and parameters (State)	2025-01-01 15:10:54 +00:00