Update Dockerfile

Dockerfile

@@ -1,4 +1,17 @@
-FROM llm-api:vllm
+FROM pytorch/pytorch:latest
+
+WORKDIR /workspace/
+ENV PYTHONPATH /workspace/
+COPY requirements.txt /workspace/
+
+#RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple && \
+RUN pip install bitsandbytes --upgrade && \
+    pip install vllm>=0.4.3 && \
+    pip install --no-cache-dir -r /workspace/requirements.txt && \
+    pip uninstall transformer-engine -y && \
+    git clone https://github.com/xusenlinzy/api-for-open-llm.git
+
+CMD python api-for-open-llm/api/server.py
 
 
 CMD python api/server.py