From 8f26f2340fb6c63bea5b37f88de3024abc092a3e Mon Sep 17 00:00:00 2001
From: Jintao Huang <huangjintao.hjt@alibaba-inc.com>
Date: Fri, 15 Aug 2025 16:01:37 +0800
Subject: [PATCH] update swift docker

---
 docker/build_image.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/docker/build_image.py b/docker/build_image.py
index 3e4fda0e..49ffd16d 100644
--- a/docker/build_image.py
+++ b/docker/build_image.py
@@ -341,6 +341,10 @@ class LLMImageBuilder(Builder):
 class SwiftImageBuilder(LLMImageBuilder):
 
     def init_args(self, args) -> Any:
+        if not args.base_image:
+            args.base_image = 'nvidia/cuda:12.6.3-devel-ubuntu22.04'
+        if not args.cuda_version:
+            args.cuda_version = '12.6.3'
         if not args.torch_version:
             args.torch_version = '2.7.1'
             args.torchaudio_version = '2.7.1'