Skip to content

Commit 49e3d83

Browse files
author
xusenlin
committed
Update requirements
1 parent 02f6b3f commit 49e3d83

File tree

3 files changed

+5
-6
lines changed

3 files changed

+5
-6
lines changed

docker/Dockerfile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,5 +6,5 @@ COPY requirements.txt /workspace/
66
COPY . /workspace
77

88
RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple && \
9-
pip install --no-cache-dir -r /workspace/requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple && \
10-
pip install bitsandbytes --upgrade -i https://pypi.tuna.tsinghua.edu.cn/simple
9+
pip install --no-cache-dir -r /workspace/requirements.txt && \
10+
pip install bitsandbytes --upgrade

docker/Dockerfile.vllm

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ COPY requirements.txt /workspace/
66
COPY . /workspace
77

88
RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple && \
9-
pip install --no-cache-dir -r /workspace/requirements.txt && \
109
pip install bitsandbytes --upgrade && \
1110
pip install vllm>=0.2.2 && \
11+
pip install --no-cache-dir -r /workspace/requirements.txt && \
1212
pip uninstall transformer-engine -y

docs/VLLM_SCRIPT.md

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,13 +12,12 @@ docker build -f docker/Dockerfile.vllm -t llm-api:vllm .
1212

1313
### 本地环境
1414

15-
安装依赖
16-
15+
安装依赖,确保安装顺序严格按照下面的命令:
1716

1817
```shell
19-
pip install -r requirements.txt
2018
pip install torch==2.1.0
2119
pip install vllm>=0.2.2
20+
pip install -r requirements.txt
2221
pip uninstall transformer-engine -y
2322
```
2423

0 commit comments

Comments
 (0)