PSHuman / requirements.txt
fffiloni's picture
Update requirements.txt
1adeed6 verified
raw
history blame
607 Bytes
torch==2.1.0
torchvision==0.16.0
torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121
diffusers==0.27.2
einops==0.8.0
gradio==5.7.1
huggingface_hub==0.25.2
imageio==2.36.0
imageio-ffmpeg==0.5.1
kornia==0.7.4
kornia_rs==0.1.7
mediapipe==0.10.18
moviepy==1.0.3
omegaconf==2.3.0
open3d==0.18.0
opencv-contrib-python==4.10.0.84
opencv-python==4.10.0.84
opencv-python-headless==4.10.0.84
peft==0.13.2
rembg==2.0.59
transformers==4.46.2
trimesh==4.5.2
triton==2.1.0
xformers==0.0.22.post7
yacs==0.1.8
git+https://github.com/facebookresearch/pytorch3d.git@75ebeeaea0908c5527e7b1e305fbc7681382db47