mirror of
https://github.com/THUDM/CogVideo.git
synced 2025-04-05 19:41:59 +08:00
a001842834
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
10 lines
235 B
Python
10 lines
235 B
Python
from ..cogvideox_t2v.lora_trainer import CogVideoXT2VLoraTrainer
|
|
from ..utils import register
|
|
|
|
|
|
class CogVideoX1dot5T2VLoraTrainer(CogVideoXT2VLoraTrainer):
|
|
pass
|
|
|
|
|
|
register("cogvideox1.5-t2v", "lora", CogVideoX1dot5T2VLoraTrainer)
|