Spaces:
Sleeping
Sleeping
Damien Benveniste
commited on
Commit
·
0ae84dd
1
Parent(s):
a8f0134
modified
Browse files- Dockerfile +6 -6
Dockerfile
CHANGED
@@ -1,6 +1,9 @@
|
|
1 |
# Use the official vLLM image as the base image
|
2 |
FROM vllm/vllm-openai:latest
|
3 |
|
|
|
|
|
|
|
4 |
# Set environment variables
|
5 |
ENV HUGGING_FACE_HUB_TOKEN="your_hf_token_here"
|
6 |
ENV HF_HOME="/tmp/huggingface"
|
@@ -10,15 +13,10 @@ ENV OUTLINES_CACHE_DIR="/tmp/outlines_cache"
|
|
10 |
ENV XDG_CONFIG_HOME="/.config"
|
11 |
ENV VLLM_USE_MODELSCOPE="false"
|
12 |
ENV VLLM_DISABLE_USAGE_STATS="true"
|
13 |
-
# ENV VLLM_API_BASE_PATH="/v1"
|
14 |
|
15 |
# Ensure PATH includes common Python locations
|
16 |
ENV PATH="/usr/local/bin:/usr/bin:/bin:/usr/local/sbin:/usr/sbin:/sbin:$PATH"
|
17 |
|
18 |
-
# Create necessary directories and set permissions
|
19 |
-
RUN mkdir -p /tmp/huggingface /tmp/cache /tmp/numba_cache /tmp/outlines_cache /.config && \
|
20 |
-
chmod -R 777 /tmp/huggingface /tmp/cache /tmp/numba_cache /tmp/outlines_cache /.config
|
21 |
-
|
22 |
# Set the working directory
|
23 |
WORKDIR /app
|
24 |
|
@@ -30,7 +28,9 @@ RUN chmod +x /app/entrypoint.sh
|
|
30 |
EXPOSE 8000
|
31 |
|
32 |
# Set the entrypoint
|
33 |
-
ENTRYPOINT ["/app/entrypoint.sh"]
|
|
|
|
|
34 |
|
35 |
# FROM python:3.9
|
36 |
|
|
|
1 |
# Use the official vLLM image as the base image
|
2 |
FROM vllm/vllm-openai:latest
|
3 |
|
4 |
+
# Install debugging tools
|
5 |
+
RUN apt-get update && apt-get install -y procps vim
|
6 |
+
|
7 |
# Set environment variables
|
8 |
ENV HUGGING_FACE_HUB_TOKEN="your_hf_token_here"
|
9 |
ENV HF_HOME="/tmp/huggingface"
|
|
|
13 |
ENV XDG_CONFIG_HOME="/.config"
|
14 |
ENV VLLM_USE_MODELSCOPE="false"
|
15 |
ENV VLLM_DISABLE_USAGE_STATS="true"
|
|
|
16 |
|
17 |
# Ensure PATH includes common Python locations
|
18 |
ENV PATH="/usr/local/bin:/usr/bin:/bin:/usr/local/sbin:/usr/sbin:/sbin:$PATH"
|
19 |
|
|
|
|
|
|
|
|
|
20 |
# Set the working directory
|
21 |
WORKDIR /app
|
22 |
|
|
|
28 |
EXPOSE 8000
|
29 |
|
30 |
# Set the entrypoint
|
31 |
+
# ENTRYPOINT ["/app/entrypoint.sh"]
|
32 |
+
COPY test_server.py /app/test_server.py
|
33 |
+
CMD ["python", "/app/test_server.py"]
|
34 |
|
35 |
# FROM python:3.9
|
36 |
|