Wan2GP_generate_video / requirements.txt
bilalRHCH's picture
Update requirements.txt
987153d verified
# Torch suite strictly pinned to match the flash-attn wheel below
torch==2.6.0
torchvision==0.21.0
torchdata==0.10.1
# Pre-compiled Flash Attention (must match Python 3.10 and Torch 2.6)
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
# Gradio strictly pinned to perfectly match your README.md sdk_version
gradio==5.33.0
# Core AI and processing libraries (loosened to >= to prevent pip resolution errors)
opencv-python>=4.9.0.80
diffusers>=0.31.0
transformers>=4.51.3
tokenizers>=0.20.3
accelerate>=1.1.1
peft>=0.17.0
mmgp>=3.4.8
rembg[gpu]>=2.0.65
# Prevent Numpy 2.x breaking changes
numpy>=1.23.5,<2
# Prevent MoviePy 2.x breaking changes (v2 completely changed the API)
moviepy==1.0.3
# Pydantic loosened to allow Gradio to use its preferred version
pydantic>=2.10.6
# Unpinned Image/Video/Audio utilities (pip will automatically find the best versions)
tqdm
imageio
easydict
ftfy
dashscope
imageio-ffmpeg
einops
mutagen
decord
onnxruntime-gpu
matplotlib
timm
segment-anything
omegaconf
hydra-core
librosa
loguru
sentencepiece
av