PosterLLaVA / requirements.txt
posterllava's picture
Update requirements.txt
d778630 verified
--extra-index-url https://download.pytorch.org/whl/cu118
pip
einops
fastapi
uvicorn
markdown2[all]
numpy
requests
sentencepiece
tokenizers==0.13.3
torch==2.0.1
torchvision==0.15.2
uvicorn
ninja
wandb
shortuuid
httpx==0.24.0
peft==0.4.0
pydantic==1.10.9
transformers==4.31.0
accelerate==0.21.0
bitsandbytes==0.41.0
scikit-learn==1.2.2
sentencepiece==0.1.99
einops==0.6.1
einops-exts==0.0.4
timm==0.6.13
typing_extensions==4.6.3
gradio==3.35.2
gradio_client==0.2.9