torch == 2.0.1
torchvision == 0.15.2
xformers == 0.0.21
numpy
regex
torchmetrics
accelerate
gsplat
open3d
tqdm
omegaconf
opencv-python
opencv-contrib-python
plyfile
timm
wandb
ftfy
pillow_heif
diffdist
diffusers
einops
imageio
imageio-ffmpeg
transformers
torchsde
huggingface-hub