requirements.txt

torch==1.12.1
torchvision==0.13.1
cudatoolkit
salesforce-lavis
bitsandbytes
git+https://github.com/huggingface/transformers.git
accelerate==0.16.0
sentence-transformers==2.2.2
sentencepiece==0.1.97