torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-lightning==2.1.2 einops omegaconf deepspeed torchmetrics webdataset accelerate tensorboard PyMCubes trimesh rembg transformers==4.34.1 diffusers==0.19.3 bitsandbytes imageio[ffmpeg] xatlas plyfile xformers==0.0.22.post7 git+https://github.com/NVlabs/nvdiffrast/ huggingface-hub accelerate==0.28.0 mesh2sdf==1.1.0 einops==0.7.0 einx==0.1.3 optimum==1.18.0