add support for arm architecture for published images (#87)

.github/workflows/cpu_image.yaml

@@ -39,6 +39,7 @@ jobs:
         with:
           context: .
           file: ./Dockerfile
+          platforms: linux/amd64, linux/arm64
           push: true
           tags: |
             ${{ env.REGISTRY }}/${{ github.repository_owner }}/${{ env.IMAGE_NAME }}:${{ github.sha }}

.github/workflows/cuda_image.yaml

@@ -71,6 +71,7 @@ jobs:
         with:
           context: .
           file: ./Dockerfile.cuda12
+          platforms: linux/amd64, linux/arm64
           push: true
           tags: |
             ${{ env.REGISTRY }}/${{ github.repository_owner }}/${{ env.CUDA_12_IMAGE_NAME }}:${{ github.sha }}

.github/workflows/gptq_image.yaml

@@ -39,6 +39,7 @@ jobs:
         with:
           context: .
           file: ./Dockerfile.gptq
+          platforms: linux/amd64, linux/arm64
           push: true
           tags: |
             ${{ env.REGISTRY }}/${{ github.repository_owner }}/${{ env.GPTQ_IMAGE_NAME }}:${{ github.sha }}

.github/workflows/metal_image.yaml

@@ -39,6 +39,7 @@ jobs:
         with:
           context: .
           file: ./Dockerfile.metal
+          platforms: linux/amd64, linux/arm64
           push: true
           tags: |
             ${{ env.REGISTRY }}/${{ github.repository_owner }}/${{ env.METAL_IMAGE_NAME }}:${{ github.sha }}

.github/workflows/smoke_test.yaml

@@ -52,6 +52,7 @@ jobs:
         with:
           context: .
           file: ./Dockerfile
+          platforms: linux/amd64, linux/arm64
           push: true
           tags: |
             ${{ env.REGISTRY }}/${{ env.REPO_ORG_NAME }}/${{ env.IMAGE_NAME }}:${{ github.sha }}
@@ -168,7 +169,7 @@ jobs:
   gpt-neox-smoke-test:
     runs-on: ubuntu-latest
     needs: build-image
-    steps:  
+    steps:
       - name: Create k8s Kind Cluster
         uses: helm/kind-action@v1.7.0
       - name: Set up Helm
@@ -184,7 +185,7 @@ jobs:
       - name: Checkout
         uses: actions/checkout@v3
         with:
-          fetch-depth: 0 
+          fetch-depth: 0
       - name: Install ialacol with gpt-neox based model and wait for pods to be ready
         run: |
           cat > values.yaml <<EOF
@@ -236,7 +237,7 @@ jobs:
   starcoder-smoke-test:
     runs-on: ubuntu-latest
     needs: build-image
-    steps:  
+    steps:
       - name: Create k8s Kind Cluster
         uses: helm/kind-action@v1.7.0
       - name: Set up Helm
@@ -252,7 +253,7 @@ jobs:
       - name: Checkout
         uses: actions/checkout@v3
         with:
-          fetch-depth: 0 
+          fetch-depth: 0
       - name: Install ialacol with starcoder based model and wait for pods to be ready
         run: |
           cat > values.yaml <<EOF
@@ -376,4 +377,3 @@ jobs:
       - if: always()
         run: |
           kubectl logs --tail=200 --selector app.kubernetes.io/name=$LLAMA_HELM_RELEASE_NAME -n $HELM_NAMESPACE
-