From 0d6402ddfda59a2074fdb089436418356c17bc4b Mon Sep 17 00:00:00 2001
From: Woosuk Kwon <woosuk.kwon@berkeley.edu>
Date: Wed, 10 Apr 2024 07:52:45 +0000
Subject: [PATCH] Fix requirements

---
 requirements-tpu.txt | 18 ++----------------
 1 file changed, 2 insertions(+), 16 deletions(-)

diff --git a/requirements-tpu.txt b/requirements-tpu.txt
index 6a11eebc547c7..624880ffa00e4 100644
--- a/requirements-tpu.txt
+++ b/requirements-tpu.txt
@@ -1,16 +1,2 @@
-psutil
-ray >= 2.9
-sentencepiece  # Required for LLaMA tokenizer.
-numpy
-torch ~= 2.2.0
-torch_xla[tpu] ~= 2.2.0
-jax[tpu]  # Required for Pallas kernels.
-requests
-py-cpuinfo
-transformers >= 4.39.1  # Required for StarCoder2 & Llava.
-fastapi
-uvicorn[standard]
-pydantic >= 2.0  # Required for OpenAI server.
-prometheus_client >= 0.18.0
-outlines == 0.0.34
-tiktoken == 0.6.0 # Required for DBRX tokenizer
+# Common dependencies
+-r requirements-common.txt