einops==0.7.0
gradio==4.10.0
numpy==1.26.2
pytorch_lightning==2.1.2
torch==2.1.2
torchmetrics==1.2.1
torchvision==0.16.2
x_transformers==1.26.2