MQT-LLaVA / requirements.txt
gordonhubackup's picture
init
140efa3
raw
history blame
404 Bytes
# --extra-index-url https://download.pytorch.org/whl/cu113
torch==2.1.2
torchvision==0.16.2
transformers==4.36.2
tokenizers==0.15.1
sentencepiece==0.1.99
shortuuid
accelerate==0.21.0
peft
pydantic
bitsandbytes
markdown2[all]
scikit-learn==1.2.2
numpy
requests
einops==0.6.1
einops-exts==0.0.4
timm==0.6.13
fastapi
# "gradio==4.16.0", "gradio_client==0.8.1",
# "", "httpx==0.24.0", "uvicorn", "",