huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-10 20:04:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

login to registry

Browse Source
This commit is contained in:
Felix Marty 2024-06-26 10:58:52 +00:00
parent 2330052aa2
commit 4067fc8211
2 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
.github/workflows/build.yaml vendored
View File

@ -168,7 +168,7 @@ jobs:
fi fi
login_tailscale: login_tailscale_and_registry:
runs-on: ["self-hosted", "${{ needs.build-and-push.outputs.runs_on }}", "multi-gpu"] runs-on: ["self-hosted", "${{ needs.build-and-push.outputs.runs_on }}", "multi-gpu"]
needs: build-and-push needs: build-and-push
concurrency: concurrency:
@ -182,10 +182,17 @@ jobs:
with: with:
authkey: ${{ secrets.TAILSCALE_AUTHKEY }} 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: prepare_integration_tests:
runs-on: ["self-hosted", "${{ needs.build-and-push.outputs.runs_on }}", "multi-gpu"] 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: concurrency:
group: ${{ github.workflow }}-${{ github.job }}-${{ needs.build-and-push.outputs.label }}-${{ github.head_ref || github.run_id }} group: ${{ github.workflow }}-${{ github.job }}-${{ needs.build-and-push.outputs.label }}-${{ github.head_ref || github.run_id }}
cancel-in-progress: true cancel-in-progress: true

2
integration-tests/clean_cache_and_download.py
View File

@ -115,7 +115,7 @@ def cleanup_cache(token: str, cache_dir: str):
print("total_required_size", total_required_size) print("total_required_size", total_required_size)
print("total_required_cached_size", total_required_cached_size) print("total_required_cached_size", total_required_cached_size)
total_non_cached_required_size = total_required_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( print(
f"Total non-cached required models size: {total_non_cached_required_size:.2f} GB (to be downloaded)" f"Total non-cached required models size: {total_non_cached_required_size:.2f} GB (to be downloaded)"