Merge branch 'dev' into feature/mpnet

kozistr · Dec 12, 2024 · 265deb2 · 265deb2
2 parents 9b58292 + 0462171
commit 265deb2
Show file tree

Hide file tree

Showing 56 changed files with 7,059 additions and 1,615 deletions.
diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
@@ -61,9 +61,7 @@ jobs:
         uses: docker/setup-buildx-action@v3
         with:
           install: true
-          buildkitd-config-inline: |
-            [registry."docker.io"]
-              mirrors = ["registry-us-east-1.prod.aws.ci.huggingface.tech"]
+          buildkitd-config: /tmp/buildkitd.toml
 
       - name: Configure sccache
         uses: actions/github-script@v6
@@ -129,6 +127,7 @@ jobs:
 
       - name: Extract metadata (tags, labels) for Docker
         id: meta-grpc
+        if: ${{ matrix.grpc }}
         uses: docker/metadata-action@v5
         with:
           images: |
@@ -144,6 +143,7 @@ jobs:
 
       - name: Build and push Docker image
         id: build-and-push-grpc
+        if: ${{ matrix.grpc }}
         uses: docker/build-push-action@v6
         with:
           context: .

diff --git a/.github/workflows/matrix.json b/.github/workflows/matrix.json
@@ -6,6 +6,7 @@
     "sccache": true,
     "cudaComputeCap": 75,
     "extraBuildArgs": "DEFAULT_USE_FLASH_ATTENTION=False",
+    "grpc": true,
     "dockerfile": "Dockerfile-cuda"
   },
   {
@@ -14,6 +15,7 @@
     "runOn": "always",
     "sccache": true,
     "cudaComputeCap": 80,
+    "grpc": true,
     "dockerfile": "Dockerfile-cuda"
   },
   {
@@ -22,6 +24,7 @@
     "runOn": "main",
     "sccache": true,
     "cudaComputeCap": 86,
+    "grpc": true,
     "dockerfile": "Dockerfile-cuda"
   },
   {
@@ -30,6 +33,7 @@
     "runOn": "main",
     "sccache": true,
     "cudaComputeCap": 89,
+    "grpc": true,
     "dockerfile": "Dockerfile-cuda"
   },
   {
@@ -38,20 +42,23 @@
     "runOn": "main",
     "sccache": true,
     "cudaComputeCap": 90,
+    "grpc": true,
     "dockerfile": "Dockerfile-cuda"
   },
   {
     "name": "All",
     "imageNamePrefix": "cuda-",
     "runOn": "main",
     "sccache": false,
+    "grpc": false,
     "dockerfile": "Dockerfile-cuda-all"
   },
   {
     "name": "cpu",
     "imageNamePrefix": "cpu-",
     "runOn": "main",
     "sccache": true,
+    "grpc": true,
     "dockerfile": "Dockerfile"
   }
 ]