minigpt_v1 / requirements.txt
saicharan1234's picture
Update requirements.txt
c7233b6 verified
raw
history blame
419 Bytes
--extra-index-url https://download.pytorch.org/whl/cu113
fastapi==0.80.0
python-multipart
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