testai / requirements.txt
MStater's picture
Upload 2 files
0dbcc5b verified
raw
history blame contribute delete
587 Bytes
# Core
huggingface_hub>=0.23
gradio>=4.44.0
imageio[ffmpeg]>=2.34
opencv-python>=4.10
einops>=0.8.0
diffsynth>=1.1.8
modelscope>=1.15.0
transformers>=4.44
sentencepiece>=0.1.99
safetensors>=0.4.3
timm>=0.9.16
accelerate>=0.33
# PyTorch (choose the wheel that matches your CUDA setup)
# For CUDA 12.4 wheels (edit for your machine or Space):
torch>=2.4.0
torchvision>=0.19.0
torchaudio>=2.4.0
# The FlashVSR repo itself installs its own python deps; we pip -e it in app.py.
# Optional speedups (compiled; only if building Block-Sparse-Attention)
packaging
ninja