"docs/zh_CN/CommunitySharings/RecommendersSvd.md" did not exist on "a1f926661689b6db3bba72f7cd381aac118e5c0a"
requirements.txt 446 Bytes
Newer Older
1
2
ninja  # For faster builds.
psutil
3
ray >= 2.9
4
5
sentencepiece  # Required for LLaMA tokenizer.
numpy
6
torch == 2.1.2
Junyang Lin's avatar
Junyang Lin committed
7
transformers >= 4.37.0 # Required for Qwen2
8
xformers == 0.0.23.post1  # Required for CUDA 12.1.
9
fastapi
10
uvicorn[standard]
11
pydantic >= 2.0  # Required for OpenAI server.
12
aioprometheus[starlette]
13
pynvml == 11.5.0
whyiug's avatar
whyiug committed
14
triton >= 2.1.0
Woosuk Kwon's avatar
Woosuk Kwon committed
15
cupy-cuda12x == 12.3.0  # Required for CUDA graphs. CUDA 11.8 users should install cupy-cuda11x instead.