diffusers==0.21.4
accelerate
transformers
xformers
basicsr
ultralytics
salesforce-lavis
webdataset
pytorch_lightning
torch==2.1.2
torchvision==0.16.2
spacy
gradio_imageslider