x

2026-03-29 04:32:37 +08:00
parent 9f128486de
commit acd5ed518e
2 changed files with 15 additions and 0 deletions
@@ -4,6 +4,9 @@ FROM docker.1ms.run/kyuz0/vllm-therock-gfx1201:latest
 ENV USE_DEFAULT_MODEL=true
 ENV LOCAL_MODEL_DIR=/opt/model

+# Ensure latest Transformers model definitions are available
+RUN /opt/venv/bin/pip install --no-cache-dir --upgrade "git+https://github.com/huggingface/transformers.git"
+
 # Create necessary directories
 RUN mkdir -p /opt/script /opt/model /config