requirements.txt

tqdm
rich
ninja
numpy
scipy
scikit-learn
matplotlib
opencv-python
imageio
imageio-ffmpeg
pandas

torch
torchvision
torch-ema
einops
tensorboard
tensorboardX

# for gui
dearpygui

# for grid_tcnn
# git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch

# for stable-diffusion
huggingface_hub
diffusers >= 0.9.0
accelerate # required by textual inversion 
transformers

# for dmtet
xatlas
PyMCubes
git+https://github.com/NVlabs/nvdiffrast/


rembg

carvekit-colab
omegaconf
pytorch-lightning
taming-transformers-rom1504
kornia
git+https://github.com/openai/CLIP.git

# for omnidata
gdown

# below is not necessary if no visualization or mesh exporting
trimesh
pymeshlab==2022.2.post4 # pointed out by @yuhuangyue in issue: https://github.com/guochengqian/Magic123/issues/45

# cubvh
git+https://github.com/ashawkey/cubvh

# for dnnutils
timm==0.6.7 # required by MiDaS
easydict
wandb
termcolor

# for shap-e
git+https://github.com/openai/shap-e

# optional
jupyterlab

# for remote debugging
debugpy-run

# for deepfloyd if
sentencepiece

# for gpu memory profiling
psutil

# for metrics
lpips