Remove einops from requirements (vllm-project#2049)

fyabc · Dec 12, 2023 · 7e1b21d · 7e1b21d
1 parent cb3f30c
commit 7e1b21d
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git 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
@@ -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.