login to registry

huggingface · Jun 26, 2024 · 4067fc8 · 4067fc8
1 parent 2330052
commit 4067fc8
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
@@ -168,7 +168,7 @@ jobs:
           fi
 
 
-  login_tailscale:
+  login_tailscale_and_registry:
     runs-on: ["self-hosted", "${{ needs.build-and-push.outputs.runs_on }}", "multi-gpu"]
     needs: build-and-push
     concurrency:
@@ -182,10 +182,17 @@ jobs:
         with:
           authkey: ${{ secrets.TAILSCALE_AUTHKEY }}
 
+      - name: Login to internal Container Registry
+        uses: docker/login-action@v3
+        with:
+          username: ${{ secrets.TAILSCALE_DOCKER_USERNAME }}
+          password: ${{ secrets.TAILSCALE_DOCKER_PASSWORD }}
+          registry: registry.internal.huggingface.tech
+
 
   prepare_integration_tests:
     runs-on: ["self-hosted", "${{ needs.build-and-push.outputs.runs_on }}", "multi-gpu"]
-    needs: [build-and-push, login_tailscale]
+    needs: [build-and-push, login_tailscale_and_registry]
     concurrency:
       group: ${{ github.workflow }}-${{ github.job }}-${{ needs.build-and-push.outputs.label }}-${{ github.head_ref || github.run_id }}
       cancel-in-progress: true

diff --git a/integration-tests/clean_cache_and_download.py b/integration-tests/clean_cache_and_download.py
@@ -115,7 +115,7 @@ def cleanup_cache(token: str, cache_dir: str):
     print("total_required_size", total_required_size)
     print("total_required_cached_size", total_required_cached_size)
     total_non_cached_required_size = total_required_size - total_required_cached_size
-    assert total_non_cached_required_size >= 0
+    assert total_non_cached_required_size >= -0.001
 
     print(
         f"Total non-cached required models size: {total_non_cached_required_size:.2f} GB (to be downloaded)"