SoSORA - Hourglass Diffusion Transformer with 1.58 bit QAT Download checkpoint Training Download and extract UCF-101 Dataset from here accelerate launch train_ucf101.py