Update requirements

Author: xusenlin

docker/Dockerfile

@@ -3,7 +3,6 @@ FROM nvcr.io/nvidia/pytorch:23.10-py3
 WORKDIR /workspace/
 ENV PYTHONPATH /workspace/
 COPY requirements.txt /workspace/
-COPY . /workspace
 
 RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple && \
     pip install --no-cache-dir -r /workspace/requirements.txt && \

docker/Dockerfile.gptq

@@ -3,9 +3,8 @@ FROM nvcr.io/nvidia/pytorch:23.10-py3
 WORKDIR /workspace/
 ENV PYTHONPATH /workspace/
 COPY requirements.txt /workspace/
-COPY . /workspace
 
 RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple && \
     pip install --no-cache-dir -r /workspace/requirements.txt && \
     pip install bitsandbytes -U && \
-    pip install auto-gptq optimum autoawq
+    pip install auto-gptq optimum

docker/Dockerfile.tgi

@@ -1,4 +1,4 @@
-FROM ghcr.io/huggingface/text-generation-inference:1.2
+FROM ghcr.io/huggingface/text-generation-inference:1.3
 
 COPY requirements.txt .
 RUN pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

docker/Dockerfile.vllm

@@ -3,10 +3,9 @@ FROM nvcr.io/nvidia/pytorch:23.10-py3
 WORKDIR /workspace/
 ENV PYTHONPATH /workspace/
 COPY requirements.txt /workspace/
-COPY . /workspace
 
 RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple && \
     pip install bitsandbytes --upgrade && \
-    pip install vllm>=0.2.2 && \
+    pip install vllm>=0.2.6 && \
     pip install --no-cache-dir -r /workspace/requirements.txt && \
     pip uninstall transformer-engine -y

docs/VLLM_SCRIPT.md

@@ -16,7 +16,7 @@ docker build -f docker/Dockerfile.vllm -t llm-api:vllm .
 
 ```shell
 pip install torch==2.1.0
-pip install vllm>=0.2.2
+pip install vllm>=0.2.6
 pip install -r requirements.txt 
 pip uninstall transformer-engine -y
 ```