Upload requirements.txt

requirements.txt

@@ -0,0 +1,38 @@
+# 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