Skip to content

GitLab

Unverified Commit 7e1b21da authored by Woosuk Kwon's avatar Woosuk Kwon Committed by GitHub
Browse files

Remove einops from requirements (#2049)

parent cb3f30c6
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
requirements-rocm.txt
View file @ 7e1b21da
...@@ -9,7 +9,6 @@ sentencepiece # Required for LLaMA tokenizer. ...@@ -9,7 +9,6 @@ sentencepiece # Required for LLaMA tokenizer.
numpy numpy
tokenizers>=0.15.0 tokenizers>=0.15.0
huggingface_hub<0.18,>=0.16.4 huggingface_hub<0.18,>=0.16.4
einops # Required for phi-1_5
transformers >= 4.36.0 # Required for Mixtral. transformers >= 4.36.0 # Required for Mixtral.
fastapi fastapi
uvicorn[standard] uvicorn[standard]
......
requirements.txt
View file @ 7e1b21da
...@@ -5,7 +5,6 @@ pandas # Required for Ray data. ...@@ -5,7 +5,6 @@ pandas # Required for Ray data.
pyarrow # Required for Ray data. pyarrow # Required for Ray data.
sentencepiece # Required for LLaMA tokenizer. sentencepiece # Required for LLaMA tokenizer.
numpy numpy
einops # Required for phi-1_5
torch >= 2.1.1 torch >= 2.1.1
transformers >= 4.36.0 # Required for Mixtral. transformers >= 4.36.0 # Required for Mixtral.
xformers >= 0.0.23 # Required for CUDA 12.1. xformers >= 0.0.23 # Required for CUDA 12.1.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment