diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index ad5e1e1d..b1fcc144 100644 --- 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 index 4fea3f0a..0be55de0 100644 --- 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)"