torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0 pytorch-lightning einops omegaconf deepspeed torchmetrics webdataset accelerate tensorboard PyMCubes trimesh rembg transformers==4.46.3 diffusers==0.32.1 bitsandbytes imageio[ffmpeg] xatlas plyfile xformers onnxruntime