Libra / requirements.txt
X-iZhang's picture
Upload 2 files
12f1914 verified
raw
history blame
463 Bytes
--extra-index-url https://download.pytorch.org/whl/cu118
pip
einops
fastapi
gradio==3.35.2
markdown2[all]
numpy
requests
sentencepiece
tokenizers>=0.12.1
torch==2.4.0
torchvision==0.19.0
uvicorn
shortuuid
httpx==0.24.0
deepspeed==0.9.5
peft==0.4.0
transformers==4.43.3
tokenizers>=0.14.0,<0.20
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
gradio_client==0.2.9
accelerate==0.33.0