requirements-rocm.txt 302 Bytes
Newer Older
1
2
3
4
ninja  # For faster builds.
typing-extensions>=4.8.0
starlette
psutil
zhuwenwen's avatar
zhuwenwen committed
5
ray == 2.9.3
6
7
8
sentencepiece  # Required for LLaMA tokenizer.
numpy
tokenizers>=0.15.0
9
transformers >= 4.38.0  # Required for Gemma.
10
11
fastapi
uvicorn[standard]
12
pydantic >= 2.0  # Required for OpenAI server.
13
prometheus_client >= 0.18.0