MagicDriveDiT
π Paper | π Website | π LICENSE | π€ GitHub
This repository contains the model checkpoint of the paper.
This model is fine-tuned for 10k steps with lr=1e-5 after 40k steps of training for stage-3.
Note:
- For inference/testing, we recommend the ema model, i.e.,
ema.pt
. - For fine-tuning, you can consider to use the training checkpoint, i.e.,
model/*
MagicDriveDiT: High-Resolution Long Video Generation for Autonomous Driving with Adaptive Control
Ruiyuan Gao1, Kai Chen2, Bo Xiao3, Lanqing Hong4, Zhenguo Li4, Qiang Xu1
1CUHK 2HKUST 3Huawei Cloud 4Huawei Noah's Ark Lab
Please find more information on our GitHub: https://github.com/flymin/MagicDriveDiT
Inference Providers
NEW
This model is not currently available via any of the supported third-party Inference Providers, and
HF Inference API was unable to determine this model's library.