mirror of
https://github.com/THUDM/CogVideo.git
synced 2025-04-05 19:41:59 +08:00
caa24bdc36
- Add SFT (Supervised Fine-Tuning) trainers for all model variants: - CogVideoX I2V and T2V - CogVideoX-1.5 I2V and T2V - Add DeepSpeed ZeRO configuration files: - ZeRO-2 with and without CPU offload - ZeRO-3 with and without CPU offload - Add base accelerate config for distributed training - Update trainer.py to support SFT training mode This enables full-parameter fine-tuning with memory-efficient distributed training using DeepSpeed ZeRO optimization.
10 lines
229 B
Python
10 lines
229 B
Python
from ..cogvideox_t2v.sft_trainer import CogVideoXT2VSftTrainer
|
|
from ..utils import register
|
|
|
|
|
|
class CogVideoX1dot5T2VSftTrainer(CogVideoXT2VSftTrainer):
|
|
pass
|
|
|
|
|
|
register("cogvideox1.5-t2v", "sft", CogVideoX1dot5T2VSftTrainer)
|