infinitetalk / requirements.txt
ShalomKing's picture
Upload requirements.txt with huggingface_hub
30a278d verified
raw
history blame contribute delete
704 Bytes
# PyTorch - must be installed first (HuggingFace Spaces handles CUDA)
--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.5.1
torchvision==0.20.1
torchaudio==2.5.1
# Flash attention (optional - may fail on some systems)
# flash-attn
# Core ML libraries
xformers==0.0.28.post3
transformers>=4.49.0
tokenizers>=0.20.3
diffusers>=0.31.0
accelerate>=1.1.1
einops
safetensors
# Gradio
gradio>=5.0.0
# Video/Image processing
opencv-python-headless>=4.9.0.80
moviepy==1.0.3
imageio
imageio-ffmpeg
scikit-image
decord
scenedetect
pillow
# Audio processing
librosa
soundfile
pyloudnorm
# Utilities
tqdm
numpy>=1.23.5,<2
easydict
ftfy
loguru
optimum-quanto==0.2.6
xfuser>=0.4.1
huggingface_hub