torch torchvision torchaudio pytorch-lightning einops omegaconf deepspeed torchmetrics webdataset accelerate tensorboard PyMCubes trimesh rembg transformers diffusers bitsandbytes imageio[ffmpeg] xatlas plyfile xformers git+https://github.com/NVlabs/nvdiffrast/ huggingface-hub numpy httpx flask pillow==9.5.0 safetensors peft sentencepiece tokenizers==0.12.1 datasets evaluate