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_i2v.lora_trainer import CogVideoXI2VLoraTrainer
|
|
from ..utils import register
|
|
|
|
|
|
class CogVideoXI2VSftTrainer(CogVideoXI2VLoraTrainer):
|
|
pass
|
|
|
|
|
|
register("cogvideox-i2v", "sft", CogVideoXI2VSftTrainer)
|