Source examples/training/finetune/Wan2.1-VSA/Wan-Syn-Data.

Finetuning Wan2.1 to make it work with VSA to accelerate#

These are e2e example scripts for finetuning Wan2.1 T2V with VSA to accelerate inference.

Execute the following commands from FastVideo/ to run training:#

Make sure you have installed VSA#

pip install vsa

Download the synthetic dataset:#

bash examples/training/finetune/Wan2.1-VSA/Wan-Syn-Data/download_dataset.sh

Slurm script to train the model#

sbatch examples/training/finetune/Wan2.1-VSA/Wan-Syn-Data/T2V-14B-VSA.slurm
sbatch examples/training/finetune/Wan2.1-VSA/Wan-Syn-Data/I2V-14B-VSA.slurm

Example materials#