Skip to content

Commit

Permalink
[CI] Update HF cache dir to use mounted storage (#1100)
Browse files Browse the repository at this point in the history
As model analysis caches all model weights, it also requires extended
disk storage. Therefore, we're updating location of HF cache as main
source of model weights.

Fix #1099
  • Loading branch information
nvukobratTT authored Jan 23, 2025
1 parent 476dd27 commit 3751526
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions .github/workflows/model-analysis.yml
Original file line number Diff line number Diff line change
Expand Up @@ -36,10 +36,12 @@ jobs:
- /etc/udev/rules.d:/etc/udev/rules.d
- /lib/modules:/lib/modules
- /opt/tt_metal_infra/provisioning/provisioning_env:/opt/tt_metal_infra/provisioning/provisioning_env
- /mnt/dockercache:/mnt/dockercache

env:
GITHUB_TOKEN: ${{ secrets.GH_TOKEN }}
HF_TOKEN: ${{ secrets.HF_TOKEN }}
HF_HOME: /mnt/dockercache/huggingface
HF_HUB_DISABLE_PROGRESS_BARS: 1

steps:
Expand Down

0 comments on commit 3751526

Please sign in to comment.