Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
requirements-rocm.txt 372 Bytes
Newer Older
TJian's avatar
Merge EmbeddedLLM/vllm-rocm into vLLM main (#1836)  
TJian committed
1
2
3
4
5
6
7
8
9
10 
ninja  # For faster builds.
typing-extensions>=4.8.0
starlette
psutil
ray >= 2.5.1
pandas  # Required for Ray data.
pyarrow  # Required for Ray data.
sentencepiece  # Required for LLaMA tokenizer.
numpy
tokenizers>=0.15.0
Woosuk Kwon's avatar
Upgrade transformers version to 4.36.0 (#2046)  
Woosuk Kwon committed
11 
transformers >= 4.36.0  # Required for Mixtral.
TJian's avatar
Merge EmbeddedLLM/vllm-rocm into vLLM main (#1836)  
TJian committed
12
13
14
15 
fastapi
uvicorn[standard]
pydantic == 1.10.13  # Required for OpenAI server.
aioprometheus[starlette]