Open-Sora-Plan icon indicating copy to clipboard operation
Open-Sora-Plan copied to clipboard

Refactor the dataset setup of videoae

Open clownrat6 opened this issue 11 months ago • 0 comments

Changed

  • Add trainer_videobase.py, dataset_videobase.py for unifying trainer and dataset of vqvae and causal vqvae.
  • Add train_ae.py for unifying training vqvae and causal vqvae.
  • Fix some name typos.

clownrat6 avatar Mar 18 '24 10:03 clownrat6