fix

Signed-off-by: oliver könig <okoenig@nvidia.com>

Author: ko3n1g

.github/actions/build-container/action.yml

@@ -24,6 +24,9 @@ inputs:
   repo-name:
     description: "The name of the repo to build container"
     required: true
+  registry:
+    description: "The registry to build container"
+    required: true
 
 runs:
   using: "composite"
@@ -61,7 +64,7 @@ runs:
       id: cache_from
       env:
         GH_TOKEN: ${{ github.token }}
-        REGISTRY: ${{ vars.ECR }}
+        REGISTRY: ${{ inputs.registry }}
       run: |
         LAST_PRS=$(gh api graphql \
           -F owner="NVIDIA-NeMo" \

.github/workflows/cicd-main.yml

@@ -155,6 +155,7 @@ jobs:
           dockerfile-path: "./docker/Dockerfile"
           PAT: ${{ secrets.PAT }}
           repo-name: "Automodel"
+          registry: ${{ vars.ECR }}
 
   cicd-unit-tests:
     strategy: