squall/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[Misc] Minor fixes in requirements.txt (#3769)

Browse Source
This commit is contained in:
Woosuk Kwon 2024-04-01 03:15:48 -07:00 committed by GitHub
parent 563c1d7ec5
commit f03cc667a0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
requirements-rocm.txt
View File

@ -15,3 +15,4 @@ uvicorn[standard]
pydantic >= 2.0 # Required for OpenAI server. pydantic >= 2.0 # Required for OpenAI server.
prometheus_client >= 0.18.0 prometheus_client >= 0.18.0
outlines == 0.0.34 outlines == 0.0.34
tiktoken == 0.6.0 # Required for DBRX tokenizer

1
requirements.txt
View File

@ -6,7 +6,6 @@ sentencepiece # Required for LLaMA tokenizer.
numpy numpy
torch == 2.1.2 torch == 2.1.2
requests requests
psutil
py-cpuinfo py-cpuinfo
transformers >= 4.39.1 # Required for StarCoder2 & Llava. transformers >= 4.39.1 # Required for StarCoder2 & Llava.
xformers == 0.0.23.post1 # Required for CUDA 12.1. xformers == 0.0.23.post1 # Required for CUDA 12.1.