torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-lightning==2.1.2 einops omegaconf deepspeed torchmetrics webdataset sentencepiece accelerate tensorboard PyMCubes trimesh rembg peft transformers==4.44.0 diffusers==0.31.0 bitsandbytes imageio[ffmpeg] xatlas plyfile xformers==0.0.22.post7 git+https://github.com/NVlabs/nvdiffrast/ huggingface-hub==0.25.2 optimum-quanto==0.2.5