--extra-index-url https://download.pytorch.org/whl/cu113
fastapi==0.80.0
torch
torchvision
salesforce-lavis
bitsandbytes
accelerate
uvicorn
torchaudio
huggingface-hub
matplotlib
psutil
iopath
pyyaml
regex
tokenizers
tqdm
transformers
timm
webdataset
omegaconf
opencv-python
decord
peft==0.2.0
sentence-transformers
scikit-image
visual-genome

wandb


git+https://github.com/huggingface/transformers.git