requirements.txt

rembg
onnxruntime-gpu==1.17
#torch==2.4.0 --index-url https://download.pytorch.org/whl/cu118
#torchvision==0.19.0 --index-url https://download.pytorch.org/whl/cu118
#torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cu118
torch==2.4.0 --index-url https://download.pytorch.org/whl/cu121
torchvision==0.19.0 --index-url https://download.pytorch.org/whl/cu121
torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cu121
accelerate==0.34.2
bpy==3.6.0
Cython==3.0.11
diffusers==0.30.3
einops==0.8.0
gin-config==0.5.0
imageio==2.35.1
imgaug==0.4.0
matplotlib==3.9.2
matplotlib-inline==0.1.7
omegaconf==2.3.0
opencv-python==4.8.0.74
opencv-python-headless==4.5.5.64
pandas==2.2.2
PyOpenGL==3.1.0
pyrender==0.1.45
PyYAML==6.0.2
regex==2024.9.11
safetensors==0.4.5
scikit-image==0.24.0
scikit-learn==1.5.1
scipy==1.14.0
six==1.16.0
tensorboard==2.18.0
tensorboard-data-server==0.7.2
timm==0.4.12
tokenizers==0.15.2
tqdm==4.66.5
transformers==4.36.0
trimesh==4.4.4
triton==3.0.0
xformers===0.0.27.post2
#https://download.pytorch.org/whl/cu118/xformers-0.0.27.post1%2Bcu118-cp310-cp310-manylinux2014_x86_64.whl
git+https://github.com/princeton-vl/[email protected]