torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 pytorch-lightning==2.2.4 einops omegaconf deepspeed torchmetrics webdataset accelerate tensorboard PyMCubes trimesh rembg transformers==4.40.2 diffusers==0.27.2 bitsandbytes imageio[ffmpeg] xatlas plyfile xformers==0.0.22.post7 git+https://github.com/NVlabs/nvdiffrast/ huggingface-hub httpx==0.23.0 flask pillow safetensors