accelerate
git+https://github.com/salesforce/lavis.git
torch