diff --git a/.github/workflows/publish_docker.yml b/.github/workflows/publish_docker.yml
index 8b2b9aa..f08fbb8 100644
--- a/.github/workflows/publish_docker.yml
+++ b/.github/workflows/publish_docker.yml
@@ -33,6 +33,14 @@ on:
         required: false
         type: string
         default: ${{ github.repository }}
+      base_image:
+        required: false
+        type: string
+        default: 'scipy-notebook'
+      base_image_version:
+        required: false
+        type: string
+        default: 'lab-4.1.5'
 
 jobs:
   publish-docker-image:
@@ -95,6 +103,8 @@ jobs:
             https_proxy=${{ inputs.http_proxy }}
             version=${{ steps.meta.outputs.version }}
             registry=${{ secrets.REGISTRY_HOST }}
+            BaseImage=${{ inputs.base_image }}
+            BaseImageVersion=${{ inputs.base_image_version }}
           context: ${{ inputs.docker_context }}
           file: ${{ inputs.dockerfile_path }}
           platforms: ${{ inputs.platforms }}