diff --git a/official-templates/stable-diffusion-webui/Dockerfile b/official-templates/stable-diffusion-webui/Dockerfile
index ea1847d..50d94cb 100644
--- a/official-templates/stable-diffusion-webui/Dockerfile
+++ b/official-templates/stable-diffusion-webui/Dockerfile
@@ -1,6 +1,6 @@
 FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
 
-ARG WEBUI_VERSION=v1.6.0
+ARG WEBUI_VERSION
 
 ENV DEBIAN_FRONTEND noninteractive
 ENV SHELL=/bin/bash
diff --git a/official-templates/stable-diffusion-webui/docker-bake.hcl b/official-templates/stable-diffusion-webui/docker-bake.hcl
index 1dd1ccd..5da8252 100644
--- a/official-templates/stable-diffusion-webui/docker-bake.hcl
+++ b/official-templates/stable-diffusion-webui/docker-bake.hcl
@@ -9,4 +9,7 @@ target "default" {
     scripts = "../../container-template"
     proxy = "../../container-template/proxy"
   }
+  args = {
+    WEBUI_VERSION = "v1.6.0"
+  }
 }