From 6cc2aba4357a334ded9e029f8ab74ce0c7f26ad4 Mon Sep 17 00:00:00 2001 From: Jintao Huang Date: Sun, 19 Apr 2026 14:19:09 +0800 Subject: [PATCH] fix --- docker/build_image.py | 1 + 1 file changed, 1 insertion(+) diff --git a/docker/build_image.py b/docker/build_image.py index 8fc1379e..a152aa12 100644 --- a/docker/build_image.py +++ b/docker/build_image.py @@ -503,6 +503,7 @@ parser.add_argument('--torchvision_version', type=str, default=None) parser.add_argument('--cuda_version', type=str, default=None) parser.add_argument('--ci_image', type=int, default=0) parser.add_argument('--torchaudio_version', type=str, default=None) +parser.add_argument('--optimum_version', type=str, default=None) parser.add_argument('--tf_version', type=str, default=None) parser.add_argument('--vllm_version', type=str, default=None) parser.add_argument('--lmdeploy_version', type=str, default=None)