diffusers==0.21.4
accelerate
transformers
xformers
basicsr
ultralytics
salesforce-lavis
webdataset