Spaces:
Running
on
L40S
Running
on
L40S
| # base dependencies, for Wan and Gradio | |
| # we don't have a flash attention binary yet for Pytorch 2.7 | |
| # so we use pytorch 2.6 | |
| torch==2.6.0 | |
| torchvision==0.21.0 | |
| torchdata==0.10.1 | |
| torchao==0.9.0 | |
| numpy>=1.26.4 | |
| # some important dependendencies | |
| ftfy | |
| # peft==0.15.1 | |
| peft | |
| # it is recommended to always use the latest version | |
| diffusers @ git+https://github.com/huggingface/diffusers.git@main | |
| flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl | |
| transformers>=4.45.2 | |
| accelerate | |
| safetensors | |
| imageio | |
| imageio-ffmpeg | |
| gradio==5.38.2 | |
| # more dependencies, specific to OmniAvatar | |
| librosa==0.10.2.post1 | |
| tqdm | |
| scipy==1.14.0 | |
| xfuser==0.4.1 | |
| einops |