Skip to content

Commit aa944e4

Browse files
authored
ci: cache embedding test model (abetlen#2250)
* ci: cache embedding test model * ci: restore previous model cache
1 parent dad5d0a commit aa944e4

1 file changed

Lines changed: 11 additions & 6 deletions

File tree

.github/workflows/test.yaml

Lines changed: 11 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,9 @@ env:
1414
RECURRENT_MODEL_FILE: mamba-130m-hf.Q2_K.gguf
1515
HYBRID_REPO_ID: tiiuae/Falcon-H1-Tiny-90M-Instruct-GGUF
1616
HYBRID_MODEL_FILE: Falcon-H1-Tiny-90M-Instruct-Q2_K.gguf
17-
MODEL_CACHE_KEY: qwen35-q8-mamba130m-q2-falconh1tiny-q2
17+
EMBEDDING_REPO_ID: CompendiumLabs/bge-small-en-v1.5-gguf
18+
EMBEDDING_MODEL_FILE: bge-small-en-v1.5-q4_k_m.gguf
19+
MODEL_CACHE_KEY: qwen35-q8-mamba130m-q2-falconh1tiny-q2-bge-small-q4
1820

1921
jobs:
2022
download-model:
@@ -26,16 +28,19 @@ jobs:
2628
python-version: "3.9"
2729
- name: Install huggingface-hub
2830
run: pip install huggingface-hub
31+
- name: Restore model cache
32+
uses: actions/cache@v5
33+
with:
34+
path: ~/.cache/huggingface/hub
35+
key: ${{ runner.os }}-model-${{ env.MODEL_CACHE_KEY }}
36+
restore-keys: |
37+
${{ runner.os }}-model-qwen35-q8-mamba130m-q2-falconh1tiny-q2
2938
- name: Download model
3039
run: |
3140
hf download ${{ env.REPO_ID }} ${{ env.MODEL_FILE }}
3241
hf download ${{ env.RECURRENT_REPO_ID }} ${{ env.RECURRENT_MODEL_FILE }}
3342
hf download ${{ env.HYBRID_REPO_ID }} ${{ env.HYBRID_MODEL_FILE }}
34-
- name: Cache model
35-
uses: actions/cache@v5
36-
with:
37-
path: ~/.cache/huggingface/hub
38-
key: ${{ runner.os }}-model-${{ env.MODEL_CACHE_KEY }}
43+
hf download ${{ env.EMBEDDING_REPO_ID }} ${{ env.EMBEDDING_MODEL_FILE }}
3944
4045
build-linux:
4146
needs: download-model

0 commit comments

Comments
 (0)