Damien Benveniste commited on
Commit
0ae84dd
·
1 Parent(s): a8f0134
Files changed (1) hide show
  1. 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