From 7e1b21daacaedb18a7001a4f5990d992209c4f78 Mon Sep 17 00:00:00 2001
From: Woosuk Kwon <woosuk.kwon@berkeley.edu>
Date: Tue, 12 Dec 2023 09:34:09 -0800
Subject: [PATCH] Remove einops from requirements (#2049)

---
 requirements-rocm.txt | 1 -
 requirements.txt      | 1 -
 2 files changed, 2 deletions(-)

diff --git a/requirements-rocm.txt b/requirements-rocm.txt
index c2e0dc3f464fd..441f018976a27 100644
--- a/requirements-rocm.txt
+++ b/requirements-rocm.txt
@@ -9,7 +9,6 @@ sentencepiece  # Required for LLaMA tokenizer.
 numpy
 tokenizers>=0.15.0
 huggingface_hub<0.18,>=0.16.4
-einops  # Required for phi-1_5
 transformers >= 4.36.0  # Required for Mixtral.
 fastapi
 uvicorn[standard]
diff --git a/requirements.txt b/requirements.txt
index 04b19b97babf4..9f8c729105d69 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,7 +5,6 @@ pandas  # Required for Ray data.
 pyarrow  # Required for Ray data.
 sentencepiece  # Required for LLaMA tokenizer.
 numpy
-einops  # Required for phi-1_5
 torch >= 2.1.1
 transformers >= 4.36.0  # Required for Mixtral.
 xformers >= 0.0.23  # Required for CUDA 12.1.