torch==2.0.1
torchvision==0.15.2
gradio>=4.13.0,<5.0.0
diffusers
transformers
accelerate>=0.20.3
omegaconf
pytorch-lightning
huggingface_hub
einops
numpy
Pillow
trimesh
xatlas
PyMCubes
git+https://github.com/openai/shap-e.git