vllm/requirements-tpu.txt
2024-04-16 17:06:35 +00:00

7 lines
145 B
Plaintext

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