Spaces:
Restarting
on
L4
Restarting
on
L4
fix: err
Browse files- Dockerfile +3 -3
Dockerfile
CHANGED
@@ -4,9 +4,9 @@ RUN pip install hf_transfer
|
|
4 |
|
5 |
# install vllm
|
6 |
#RUN pip install vllm
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
|
11 |
# install node and npm for demo
|
12 |
ENV NVM_DIR /usr/local/nvm
|
|
|
4 |
|
5 |
# install vllm
|
6 |
#RUN pip install vllm
|
7 |
+
ENV PIP_EXTRA_INDEX_URL https://download.pytorch.org/whl/cpu
|
8 |
+
ENV VLLM_TARGET_DEVICE cpu
|
9 |
+
RUN pip install https://github.com/vllm-project/vllm
|
10 |
|
11 |
# install node and npm for demo
|
12 |
ENV NVM_DIR /usr/local/nvm
|