xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-09-06 05:56:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add JAX to requirements.txt

Browse Source
This commit is contained in:
Woosuk Kwon 2024-04-01 08:23:59 +00:00
parent 38e3d33a62
commit 6894d3efef
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
requirements-tpu.txt
View File

@ -4,6 +4,7 @@ sentencepiece # Required for LLaMA tokenizer.
numpy numpy
torch ~= 2.2.0 torch ~= 2.2.0
torch_xla[tpu] ~= 2.2.0 torch_xla[tpu] ~= 2.2.0
jax[tpu] # Required for Pallas kernels.
requests requests
py-cpuinfo py-cpuinfo
transformers >= 4.39.1 # Required for StarCoder2 & Llava. transformers >= 4.39.1 # Required for StarCoder2 & Llava.