ci: fix flux release

Signed-off-by: Sertac Ozercan <sozercan@gmail.com>
sozercan · Sep 7, 2024 · ebf3189 · ebf3189
1 parent 1b09566
commit ebf3189
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/.github/workflows/update-models.yaml b/.github/workflows/update-models.yaml
@@ -94,13 +94,18 @@ jobs:
             export REGISTRY=ghcr.io/sozercan
           fi
 
+          export PLATFORMS="linux/amd64,linux/arm64"
+          if [ "${MODEL_NAME}" == "flux-1-dev" ]; then
+            export PLATFORMS="linux/amd64"
+          fi
+
           docker buildx build . \
             -t ${REGISTRY}/${MODEL_NAME}:${MODEL_SIZE} \
             -t ${REGISTRY}/${MODEL_NAME}:${MODEL_SIZE}${MODEL_TYPE} \
             -f models/${{ matrix.model }}.yaml \
             --push --progress plain \
             --sbom=true --provenance=true \
-            --platform linux/amd64,linux/arm64
+            --platform ${PLATFORMS}
           echo "DIGEST=$(cosign triangulate ${REGISTRY}/${MODEL_NAME}:${MODEL_SIZE} --type digest)" >> $GITHUB_ENV
 
       - name: Sign the images with GitHub OIDC Token