VCoder / requirements.txt
praeclarumjj3's picture
:zap: Fix req
4b26417
raw
history blame
440 Bytes
--extra-index-url https://download.pytorch.org/whl/cu117
torch==2.0.0+cu117
packaging
Pillow
huggingface_hub
matplotlib
flash-attn
gradio
fastapi
numpy
requests
sentencepiece
tokenizers>=0.12.1
uvicorn
chardet
shortuuid
httpx==0.24.0,
spacy
inflect
peft==0.4.0
num2words
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
gradio_client==0.2.9