From 6d62e4c6aad9aa82b0fa525a13e8b07c2a11de3c Mon Sep 17 00:00:00 2001
From: Woosuk Kwon <woosuk.kwon@berkeley.edu>
Date: Tue, 16 Apr 2024 17:06:35 +0000
Subject: [PATCH] Add torch to dependencies

---
 requirements-tpu.txt | 1 +
 1 file changed, 1 insertion(+)

diff --git a/requirements-tpu.txt b/requirements-tpu.txt
index e627bb9b9e6b5..395e362bba3eb 100644
--- a/requirements-tpu.txt
+++ b/requirements-tpu.txt
@@ -1,5 +1,6 @@
 # Common dependencies
 -r requirements-common.txt
 
+torch
 jax[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
 flax >= 0.8