torch
torchvision
salesforce-lavis
transformers==4.28.1
accelerate==0.16.0
sentence-transformers==2.2.2
sentencepiece==0.1.97