Merge pull request #5 from koyeb/add-cache

Configure caching for all build steps
koyeb · Jan 13, 2025 · 8f065c9 · 8f065c9
2 parents 37349c5 + 103b4d5
commit 8f065c9
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/.github/workflows/main.yaml b/.github/workflows/main.yaml
@@ -48,6 +48,12 @@ jobs:
         with:
           username: ${{ secrets.DOCKER_USERNAME }}
           password: ${{ secrets.DOCKER_PASSWORD}}
+      - name: Login to Koyeb cache
+        uses: docker/login-action@v3
+        with:
+          registry: registry01.prod.koyeb.com/internal-example-vllm-cache-koyeb
+          username: ${{ secrets.KOYEB_CACHE_REGISTRY_USERNAME }}
+          password: ${{ secrets.KOYEB_CACHE_REGISTRY_PASSWORD }}
       - name: Build and push
         uses: docker/build-push-action@v6
         with:
@@ -61,3 +67,7 @@ jobs:
             MODEL_NAME=${{ matrix.model.name }}
           secrets: |
             "hf_token=${{ secrets.HUGGINGFACE_TOKEN }}"
+          cache-to: |
+            type=registry,image-manifest=true,oci-mediatypes=true,ref=registry01.prod.koyeb.com/internal-example-vllm-cache-${{ matrix.model.tag }}
+          cache-from: |
+            type=registry,ref=registry01.prod.koyeb.com/internal-example-vllm-cache-${{ matrix.model.tag }}