Skip to content

Commit e83dfb3

Browse files
authored
Fix hf vllm image configs for release (#6313)
Signed-off-by: Sally Seok <sallyseo@amazon.com>
1 parent a92a65f commit e83dfb3

2 files changed

Lines changed: 3 additions & 4 deletions

File tree

.github/config/image/huggingface-vllm/sagemaker.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,8 @@ build:
2929

3030
release:
3131
release: true
32-
force_release: true
32+
force_release: false
3333
public_registry: false
3434
private_registry: true
35-
enable_soci: true
35+
enable_soci: false
3636
environment: "gamma"

docker/huggingface/vllm/Dockerfile

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,11 +6,9 @@ ARG FRAMEWORK=huggingface-vllm
66
ARG FRAMEWORK_VERSION=0.22.1
77
ARG CONTAINER_TYPE=general
88
ARG DLC_MAJOR_VERSION=1
9-
ARG DLC_MINOR_VERSION=0
109

1110
LABEL maintainer="Hugging Face"
1211
LABEL dlc_major_version="${DLC_MAJOR_VERSION}"
13-
LABEL dlc_minor_version="${DLC_MINOR_VERSION}"
1412

1513
ARG TRANSFORMERS_VERSION=5.10.2
1614
ARG HUGGINGFACE_HUB_VERSION=1.17.0
@@ -27,6 +25,7 @@ RUN uv pip install --system \
2725
"grpcio>=1.79.3" \
2826
"cryptography>=48.0.1" \
2927
"starlette>=1.3.1" \
28+
"aiohttp>=3.14.1" \
3029
&& uv cache clean \
3130
&& rm -rf /tmp/uv*
3231

0 commit comments

Comments
 (0)