We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cb3f30c commit 7e1b21dCopy full SHA for 7e1b21d
requirements-rocm.txt
@@ -9,7 +9,6 @@ sentencepiece # Required for LLaMA tokenizer.
9
numpy
10
tokenizers>=0.15.0
11
huggingface_hub<0.18,>=0.16.4
12
-einops # Required for phi-1_5
13
transformers >= 4.36.0 # Required for Mixtral.
14
fastapi
15
uvicorn[standard]
requirements.txt
@@ -5,7 +5,6 @@ pandas # Required for Ray data.
5
pyarrow # Required for Ray data.
6
sentencepiece # Required for LLaMA tokenizer.
7
8
torch >= 2.1.1
xformers >= 0.0.23 # Required for CUDA 12.1.
0 commit comments