hitxujian / sosora

Hourglass DiT video diffusion model

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

SoSORA - Hourglass Diffusion Transformer with 1.58 bit QAT

Download checkpoint

Horse race class conditional generation

Training

Download and extract UCF-101 Dataset from here

accelerate launch train_ucf101.py

About

Hourglass DiT video diffusion model

License:MIT License


Languages

Language:Python 100.0%