Safetensors
SpiritSight-Agent-2B / requirements.txt
huangzhiyuan's picture
first commit
a197764
torch==2.0.1
torchvision==0.15.2
transformers==4.37.2
tokenizers==0.15.1
sentencepiece==0.1.99
shortuuid
accelerate
peft==0.10.0
bitsandbytes==0.41.0
pydantic<2,>=1
markdown2[all]
numpy<2
scikit-learn>=1.2.2
gradio==3.35.2
gradio_client==0.2.9
requests
httpx==0.24.0
uvicorn
fastapi
deepspeed>=0.9.5
einops==0.6.1
einops-exts==0.0.4
timm>=0.6.11