Changed python location and transformer version
Browse files- 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
|
| 13 |
# plus minimal utils. vLLM image already includes CUDA/PyTorch.
|
| 14 |
-
RUN pip install --no-cache-dir "transformers
|
| 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
|
| 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"
|