tommulder commited on
Commit
aa1ab1e
·
1 Parent(s): df43814

Changed python location and transformer version

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -3
Dockerfile CHANGED
@@ -9,14 +9,14 @@ ENV HF_HUB_ENABLE_HF_TRANSFER=1 \
9
  GIT_LFS_SKIP_SMUDGE=1 \
10
  PIP_DISABLE_PIP_VERSION_CHECK=1
11
 
12
- # Install exact Transformers version dots.ocr expects (qwen2_vl APIs),
13
  # plus minimal utils. vLLM image already includes CUDA/PyTorch.
14
- RUN pip install --no-cache-dir "transformers==4.48.3" "huggingface_hub>=0.24.0"
15
 
16
  # Pre-download the model repo into /weights/DotsOCR
17
  # Note: dots.ocr requires the directory name to avoid '.' (see model card).
18
  ARG MODEL_ID=rednote-hilab/dots.ocr
19
- RUN python - <<'PY'
20
  from huggingface_hub import snapshot_download
21
  import os, shutil
22
  root = "/weights"
 
9
  GIT_LFS_SKIP_SMUDGE=1 \
10
  PIP_DISABLE_PIP_VERSION_CHECK=1
11
 
12
+ # Install compatible Transformers version for vLLM 0.9.1 (>=4.51.1 required)
13
  # plus minimal utils. vLLM image already includes CUDA/PyTorch.
14
+ RUN pip install --no-cache-dir "transformers>=4.51.1" "huggingface_hub>=0.24.0"
15
 
16
  # Pre-download the model repo into /weights/DotsOCR
17
  # Note: dots.ocr requires the directory name to avoid '.' (see model card).
18
  ARG MODEL_ID=rednote-hilab/dots.ocr
19
+ RUN python3 - <<'PY'
20
  from huggingface_hub import snapshot_download
21
  import os, shutil
22
  root = "/weights"