FaceLift / requirements.txt
Weijie Lyu
Fix real OOM: 4-D SDPA layout; cu130 is ZeroGPU-native, drop cu128 cruft
d0d5411
Raw
History Blame
795 Bytes
# torch 2.11.0 is ZeroGPU-supported and the version line where torchvision 0.26.0
# and xformers 0.0.35 agree (xformers 0.0.33 hard-pinned torch==2.9.0, which ZeroGPU
# rejects; 0.0.35 relaxed to >=2.10). ZeroGPU provides the cu130 build + a matching
# CUDA 13.0 toolkit, so diff-gaussian-rasterization compiles against the system toolkit.
torch==2.11.0
torchvision==0.26.0
numpy==1.26.4
pillow==10.4.0
diffusers[torch]==0.30.3
transformers==4.44.2
accelerate==0.33.0
opencv-python==4.10.0.84
rembg
facenet-pytorch
einops==0.8.0
ninja
plyfile==1.0.3
easydict==1.13
pyyaml==6.0.2
lpips==0.1.4
huggingface-hub==0.35.3
videoio==0.3.0
gradio==5.49.1
matplotlib==3.7.5
onnxruntime>=1.15.0
xformers==0.0.35
scikit-image==0.21.0
jaxtyping==0.2.19
pytorch-msssim==1.0.0
ffmpeg-python==0.2.0
tqdm
spaces