diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml
index cfb165e..a0f605d 100644
--- a/.github/workflows/docker.yml
+++ b/.github/workflows/docker.yml
@@ -37,6 +37,11 @@ on:
         required: false
         default: .
         type: string
+      artifact_per_platform:
+        description: "If true, downloads artifact_name-<platform> instead of artifact_name"
+        required: false
+        default: false
+        type: boolean
       push:
         description: "Whether to push the image, or just build"
         required: true
@@ -94,7 +99,7 @@ jobs:
         if: ${{ inputs.artifact_name != '' }}
         uses: actions/download-artifact@v4
         with:
-          name: ${{inputs.artifact_name}}
+          name: ${{ inputs.artifact_per_platform && format('{0}-{1}', inputs.artifact_name, matrix.platform) || inputs.artifact_name }}
           path: ${{inputs.artifact_path}}
 
       - name: Log into registry ${{ inputs.registry }}