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

Add torch to dependencies

Browse Source
This commit is contained in:
Woosuk Kwon 2024-04-16 17:06:35 +00:00
parent de82e95787
commit 6d62e4c6aa
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

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