Unique3D / requirements.txt
Wuvin's picture
x
b898902
raw
history blame
713 Bytes
pytorch3d @ https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu121_pyt221/pytorch3d-0.7.6-cp310-cp310-linux_x86_64.whl
ort_nightly_gpu # ==1.17.0.dev20240118002 --index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ort-cuda-12-nightly/pypi/simple/
onnxruntime_gpu==1.17.0 # --index-url https://pkgs.dev.azure.com/onnxruntime/onnxruntime/_packaging/onnxruntime-cuda-12/pypi/simple/
accelerate
datasets
diffusers>=0.26.3
fire
gradio
jaxtyping
numba
numpy
git+https://github.com/NVlabs/nvdiffrast.git
omegaconf>=2.3.0
opencv_python
opencv_python_headless
peft
Pillow
pygltflib
pymeshlab>=2023.12
rembg
torch>=2.0.1
torch_scatter
tqdm
transformers
trimesh
typeguard
wandb
xformers