torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-lightning==2.1.2 numpy einops omegaconf torchmetrics webdataset accelerate tensorboard PyMCubes trimesh rembg transformers==4.34.1 diffusers==0.19.3 imageio[ffmpeg] xatlas plyfile xformers==0.0.22.post7 huggingface-hub open3d