Skip to content

Commit fc49c76

Browse files
authored
Merge branch 'master' into patch
2 parents ead7766 + 9a94c90 commit fc49c76

File tree

2 files changed

+16
-4
lines changed

2 files changed

+16
-4
lines changed

huggingface/vllm/buildspec.yml

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -42,9 +42,7 @@ images:
4242
tag_python_version: &TAG_PYTHON_VERSION py312
4343
os_version: &OS_VERSION ubuntu22.04
4444
transformers_version: &TRANSFORMERS_VERSION 4.57.3
45-
vllm_version: &VLLM_VERSION 0.14.0
46-
tag: !join [ "vllm", "-", *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION, "-", *OS_VERSION, "-sagemaker" ]
47-
latest_release_tag: !join [ "vllm", "-", *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION, "-", *OS_VERSION, "-sagemaker" ]
45+
tag: !join [ *VERSION, '-', 'transformers', *TRANSFORMERS_VERSION, '-', *DEVICE_TYPE, '-', *TAG_PYTHON_VERSION, '-', *CUDA_VERSION, '-', *OS_VERSION ]
4846
docker_file: !join [ docker/, *SHORT_VERSION, /, *CUDA_VERSION, /Dockerfile ]
4947
target: sagemaker
5048
build: true
@@ -53,4 +51,4 @@ images:
5351
test_platforms:
5452
- sanity
5553
- security
56-
- sagemaker
54+
- sagemaker

release_images_general.yml

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,3 +176,17 @@ release_images:
176176
force_release: False
177177
public_registry: True
178178
enable_soci: True
179+
13:
180+
framework: "huggingface_vllm"
181+
version: "0.14.0"
182+
arch_type: "x86"
183+
hf_transformers: "4.57.3"
184+
vllm_version: "0.14.0"
185+
general: # vLLM is categorized as general type
186+
device_types: [ "gpu" ]
187+
python_versions: [ "py312" ]
188+
os_version: "ubuntu22.04"
189+
cuda_version: "cu129"
190+
example: False
191+
disable_sm_tag: True
192+
force_release: False

0 commit comments

Comments
 (0)