requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu117
torch==1.13.1+cu117
torchvision==0.14.1+cu117
torchaudio==0.13.1 

transformers==4.30.2
faiss-gpu==1.7.2
timm==0.4.12
openai
open_clip_torch
omegaconf
h5py>=3.8.0
spacy>=3.5.0
bitsandbytes
accelerate
scipy