diff --git a/Dockerfile b/Dockerfile
index f7fdbb8..beee4ea 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -3,7 +3,7 @@ ARG GO_VERSION=1.20
 FROM golang:$GO_VERSION as builder
 
 ARG BUILD_TYPE=
-ARG GO_TAGS=
+ARG GO_TAGS=stablediffusion
 ARG CUDA_MAJOR_VERSION=11
 ARG CUDA_MINOR_VERSION=7
 
@@ -45,7 +45,7 @@ RUN make build
 FROM golang:$GO_VERSION
 
 ARG BUILD_TYPE=
-ARG GO_TAGS=
+ARG GO_TAGS=stablediffusion
 ARG CUDA_MAJOR_VERSION=11
 ARG CUDA_MINOR_VERSION=7