--extra-index-url https://download.pytorch.org/whl/cu121

torch==2.2.2
torchvision==0.17.2
pytorch-lightning==2.4.0
lightning==2.3.3
matplotlib
numpy==1.26.4
gradio==5.34.2
wandb
omegaconf
imageio
diffusers
plotly
pybullet
pyrender
trimesh
numpy-quaternion
openai
spaces
json_repair