mirror of
https://github.com/THUDM/CogVideo.git
synced 2025-04-05 03:04:56 +08:00
fdb9820949
- Add DeepSpeed ZeRO-3 configuration support - Optimize memory usage during training - Rename training scripts to reflect ZeRO usage - Update related configuration files and trainers
10 lines
219 B
Python
10 lines
219 B
Python
from ..cogvideox_t2v.lora_trainer import CogVideoXT2VLoraTrainer
|
|
from ..utils import register
|
|
|
|
|
|
class CogVideoXT2VSftTrainer(CogVideoXT2VLoraTrainer):
|
|
pass
|
|
|
|
|
|
register("cogvideox-t2v", "sft", CogVideoXT2VSftTrainer)
|