mirror of
https://github.com/THUDM/CogVideo.git
synced 2025-06-26 20:29:24 +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 ..utils import register
|
|
from ..cogvideox_i2v.lora_trainer import CogVideoXI2VLoraTrainer
|
|
|
|
|
|
class CogVideoX1dot5I2VLoraTrainer(CogVideoXI2VLoraTrainer):
|
|
pass
|
|
|
|
|
|
register("cogvideox1.5-i2v", "lora", CogVideoX1dot5I2VLoraTrainer)
|