Skip to content

Commit 01e32fc

Browse files
authored
[fix][djl][tensorrt] add missing cuda_version parameter to djl trt images (#4511)
1 parent 6ac1b7e commit 01e32fc

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

release_images_inference.yml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -129,6 +129,7 @@ release_images:
129129
python_versions: [ "py310" ]
130130
os_version: "ubuntu22.04"
131131
tensorrtllm_version: "0.11.0"
132+
cuda_version: "cu124"
132133
example: False
133134
disable_sm_tag: True
134135
force_release: False
@@ -141,6 +142,7 @@ release_images:
141142
python_versions: [ "py310" ]
142143
os_version: "ubuntu22.04"
143144
tensorrtllm_version: "0.12.0"
145+
cuda_version: "cu125"
144146
example: False
145147
disable_sm_tag: True
146148
force_release: False

0 commit comments

Comments
 (0)