SkyReels-A2: Compose anything in video diffusion transformers
711
stars
62
commits
Python
primary language
Jun 3, 2025
updated
🔥 For more results, visit our homepage 🔥
👋 Join our Discord
This repo, named SkyReels-A2, contains the official PyTorch implementation of our paper SkyReels-A2: Compose Anything in Video Diffusion Transformers.
| Models | Download Link | Video Size |
|---|---|---|
| A2-Wan2.1-14B-Preview | Huggingface 🤗 | ~ 81 x 480 x 832 |
| A2-Wan2.1-14B | To be released | ~ 81 x 480 x 832 |
| A2-Wan2.1-14B-Pro | To be released | ~ 97 x 544 x 960 |
| A2-Wan2.1-14B-Infinity | To be released | ~ Inf x 720 x 1080 |
First git clone the repository with code:
git clone https://github.com/SkyworkAI/SkyReels-A2.git
cd SkyReels-A2
# create env using conda
conda create -n skyreels-a2 python=3.10
conda activate skyreels-a2
Then, install the remaining dependencies:
pip install -r requirements.txt
You can download the pretrained weights from HuggingFace as:
# !pip install -U "huggingface_hub[cli]"
huggingface-cli download Skywork/SkyReels-A2 --local-dir local_path --exclude "*.git*" "README.md" "docs"
or download from webpage mannually.
You can first set the model path and reference images path and then simply run the inference scripts as:
python infer.py
If the script runs successfully, you will get an output mp4 file. This file includes the following results: driving video, input image or video, and generated result.
We also support multi-GPU inference scripts for faster inference, as:
torchrun --nproc_per_node=$GPU_NUM infer_MGPU.py
Set the offload_switch of infer_MGPU.py to True, and you can run it on RTX4090
We also provide a Gradio interface for a better user experience, just run by:
python app.py
The graphical interactive interface is shown as below.
We public the evaluation data in Huggingface, you can infer with results and then submit to leaderboard to obtain the results automatically. More detail about metric computation code will coming soon.
We would like to thank the contributors of Wan and finetrainers repositories, for their open research and contributions.
If you find SkyReels-A2 useful for your research, welcome to 🌟 this repo and cite our work using the following BibTeX:
@article{fei2025skyreels,
title={SkyReels-A2: Compose Anything in Video Diffusion Transformers},
author={Fei, Zhengcong and Li, Debang and Qiu, Di and Wang, Jiahua and Dou, Yikun and Wang, Rui and Xu, Jingtao and Fan, Mingyuan and Chen, Guibin and Li, Yang and others},
journal={arXiv preprint arXiv:2504.02436},
year={2025}
}
Python
100.0%
SkyReels-A2: Compose anything in video diffusion transformers
711
stars
62
commits
Python
primary language
Jun 3, 2025
updated
🔥 For more results, visit our homepage 🔥
👋 Join our Discord
This repo, named SkyReels-A2, contains the official PyTorch implementation of our paper SkyReels-A2: Compose Anything in Video Diffusion Transformers.
| Models | Download Link | Video Size |
|---|---|---|
| A2-Wan2.1-14B-Preview | Huggingface 🤗 | ~ 81 x 480 x 832 |
| A2-Wan2.1-14B | To be released | ~ 81 x 480 x 832 |
| A2-Wan2.1-14B-Pro | To be released | ~ 97 x 544 x 960 |
| A2-Wan2.1-14B-Infinity | To be released | ~ Inf x 720 x 1080 |
First git clone the repository with code:
git clone https://github.com/SkyworkAI/SkyReels-A2.git
cd SkyReels-A2
# create env using conda
conda create -n skyreels-a2 python=3.10
conda activate skyreels-a2
Then, install the remaining dependencies:
pip install -r requirements.txt
You can download the pretrained weights from HuggingFace as:
# !pip install -U "huggingface_hub[cli]"
huggingface-cli download Skywork/SkyReels-A2 --local-dir local_path --exclude "*.git*" "README.md" "docs"
or download from webpage mannually.
You can first set the model path and reference images path and then simply run the inference scripts as:
python infer.py
If the script runs successfully, you will get an output mp4 file. This file includes the following results: driving video, input image or video, and generated result.
We also support multi-GPU inference scripts for faster inference, as:
torchrun --nproc_per_node=$GPU_NUM infer_MGPU.py
Set the offload_switch of infer_MGPU.py to True, and you can run it on RTX4090
We also provide a Gradio interface for a better user experience, just run by:
python app.py
The graphical interactive interface is shown as below.
We public the evaluation data in Huggingface, you can infer with results and then submit to leaderboard to obtain the results automatically. More detail about metric computation code will coming soon.
We would like to thank the contributors of Wan and finetrainers repositories, for their open research and contributions.
If you find SkyReels-A2 useful for your research, welcome to 🌟 this repo and cite our work using the following BibTeX:
@article{fei2025skyreels,
title={SkyReels-A2: Compose Anything in Video Diffusion Transformers},
author={Fei, Zhengcong and Li, Debang and Qiu, Di and Wang, Jiahua and Dou, Yikun and Wang, Rui and Xu, Jingtao and Fan, Mingyuan and Chen, Guibin and Li, Yang and others},
journal={arXiv preprint arXiv:2504.02436},
year={2025}
}
Python
100.0%