Vedio-generation / requirements.txt
Starlucas0201's picture
Update requirements.txt
9858241 verified
raw
history blame contribute delete
665 Bytes
# Core
torch
torchsde
torchvision
torchaudio
numpy==1.26.3
einops
transformers>=4.28.1
tokenizers>=0.13.3
sentencepiece
safetensors>=0.4.2
aiohttp>=3.11.8
yarl>=1.18.0
pyyaml
Pillow
scipy
tqdm
psutil
pydantic==2.10.6
# Optional & Custom
kornia>=0.7.1
spandrel
soundfile
av
opencv-python-headless>=4.7.0.72 # Use headless version for Hugging Face
imageio-ffmpeg
albumentations>=1.4.16
insightface==0.7.3
onnx>=1.14.0
onnxruntime==1.19.2 # Keep this, drop onnxruntime-gpu
# ML Tools
segment_anything
ultralytics
# Utilities
py-cpuinfo
gitpython
moviepy
scikit-build
diskcache
# Hugging Face Ecosystem
huggingface_hub
diffusers>=0.31.0
accelerate>=0.33.0
peft