pytorch-lightning==2.1.2 gradio huggingface-hub einops omegaconf torchmetrics webdataset accelerate tensorboard PyMCubes trimesh rembg transformers==4.34.1 diffusers==0.20.2 bitsandbytes imageio[ffmpeg] xatlas plyfile git+https://github.com/NVlabs/nvdiffrast/