tommulder commited on
Commit
e612421
·
1 Parent(s): b455369

User errors

Browse files
Files changed (1) hide show
  1. Dockerfile +9 -1
Dockerfile CHANGED
@@ -7,6 +7,13 @@ RUN useradd -m -u 1000 user
7
  # Switch to the "user" user
8
  USER user
9
 
 
 
 
 
 
 
 
10
  # Speed up HF downloads and avoid interactive git prompts
11
  ENV HF_HUB_ENABLE_HF_TRANSFER=1 \
12
  GIT_LFS_SKIP_SMUDGE=1 \
@@ -14,7 +21,8 @@ ENV HF_HUB_ENABLE_HF_TRANSFER=1 \
14
 
15
  # Install compatible Transformers version for vLLM 0.9.1 (>=4.51.1 required)
16
  # plus minimal utils. vLLM image already includes CUDA/PyTorch.
17
- RUN pip install --no-cache-dir "transformers>=4.51.1" "huggingface_hub>=0.24.0"
 
18
 
19
  # Pre-download the model repo into /weights/DotsOCR
20
  # Note: dots.ocr requires the directory name to avoid '.' (see model card).
 
7
  # Switch to the "user" user
8
  USER user
9
 
10
+ # Set home to the user's home directory and update PATH
11
+ ENV HOME=/home/user \
12
+ PATH=/home/user/.local/bin:$PATH
13
+
14
+ # Set the working directory to the user's home directory
15
+ WORKDIR $HOME/app
16
+
17
  # Speed up HF downloads and avoid interactive git prompts
18
  ENV HF_HUB_ENABLE_HF_TRANSFER=1 \
19
  GIT_LFS_SKIP_SMUDGE=1 \
 
21
 
22
  # Install compatible Transformers version for vLLM 0.9.1 (>=4.51.1 required)
23
  # plus minimal utils. vLLM image already includes CUDA/PyTorch.
24
+ RUN pip install --no-cache-dir --upgrade pip && \
25
+ pip install --no-cache-dir "transformers>=4.51.1" "huggingface_hub>=0.24.0"
26
 
27
  # Pre-download the model repo into /weights/DotsOCR
28
  # Note: dots.ocr requires the directory name to avoid '.' (see model card).