Damien Benveniste commited on
Commit
d0b9a2d
·
1 Parent(s): a3789d1
Files changed (2) hide show
  1. Dockerfile +3 -0
  2. entrypoint.sh +1 -1
Dockerfile CHANGED
@@ -8,6 +8,9 @@ ENV XDG_CACHE_HOME="/tmp/cache"
8
  ENV NUMBA_CACHE_DIR="/tmp/numba_cache"
9
  ENV OUTLINES_CACHE_DIR="/tmp/outlines_cache"
10
 
 
 
 
11
  # Create necessary directories and set permissions
12
  RUN mkdir -p /tmp/huggingface /tmp/cache /tmp/numba_cache /tmp/outlines_cache /tmp/config && \
13
  chmod -R 777 /tmp/huggingface /tmp/cache /tmp/numba_cache /tmp/outlines_cache /tmp/config
 
8
  ENV NUMBA_CACHE_DIR="/tmp/numba_cache"
9
  ENV OUTLINES_CACHE_DIR="/tmp/outlines_cache"
10
 
11
+ # Ensure PATH includes common Python locations
12
+ ENV PATH="/usr/local/bin:/usr/bin:/bin:/usr/local/sbin:/usr/sbin:/sbin:$PATH"
13
+
14
  # Create necessary directories and set permissions
15
  RUN mkdir -p /tmp/huggingface /tmp/cache /tmp/numba_cache /tmp/outlines_cache /tmp/config && \
16
  chmod -R 777 /tmp/huggingface /tmp/cache /tmp/numba_cache /tmp/outlines_cache /tmp/config
entrypoint.sh CHANGED
@@ -25,7 +25,7 @@ done
25
 
26
 
27
  # Construct the command
28
- CMD="python -m vllm.entrypoints.openai.api_server \
29
  --model $MODEL \
30
  --host 0.0.0.0 \
31
  --port 8000 \
 
25
 
26
 
27
  # Construct the command
28
+ CMD="vllm serve \
29
  --model $MODEL \
30
  --host 0.0.0.0 \
31
  --port 8000 \