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

Add JAX requirements

Browse Source
This commit is contained in:
Woosuk Kwon 2024-04-16 08:05:54 +00:00
parent c59c1e7b2c
commit eb0a0466a9
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
requirements-tpu.txt
View File

@ -1,2 +1,5 @@
# Common dependencies # Common dependencies
-r requirements-common.txt -r requirements-common.txt
jax[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
flax >= 0.8