OleehyO a001842834 feat: implement CogVideoX trainers for I2V and T2V tasks ...

Add and refactor trainers for CogVideoX model variants:
- Implement CogVideoXT2VLoraTrainer for text-to-video generation
- Refactor CogVideoXI2VLoraTrainer for image-to-video generation

Both trainers support LoRA fine-tuning with proper handling of:
- Model components loading and initialization
- Video encoding and batch collation
- Loss computation with noise prediction
- Validation step for generation

2025-01-01 15:10:54 +00:00

..

lora_trainer.py

feat: implement CogVideoX trainers for I2V and T2V tasks

2025-01-01 15:10:54 +00:00