File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -192,6 +192,20 @@ jobs:
192192 echo "=== lightspeed-stack.yaml ==="
193193 grep -A 3 "llama_stack:" lightspeed-stack.yaml
194194
195+ - name : Cache HuggingFace embedding model
196+ uses : actions/cache@v4
197+ with :
198+ path : /tmp/hf-cache
199+ key : hf-sentence-transformers-all-mpnet-base-v2
200+
201+ - name : Pre-download HuggingFace embedding model
202+ env :
203+ HF_HOME : /tmp/hf-cache
204+ run : |
205+ pip install -q sentence-transformers
206+ python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('sentence-transformers/all-mpnet-base-v2')"
207+ echo "HF_CACHE_PATH=/tmp/hf-cache" >> $GITHUB_ENV
208+
195209 - name : Docker Login for quay access
196210 env :
197211 QUAY_ROBOT_USERNAME : ${{ secrets.QUAY_DOWNSTREAM_USERNAME }}
Original file line number Diff line number Diff line change @@ -129,6 +129,20 @@ jobs:
129129 echo "$BODY"
130130 [ "$HTTP_CODE" = "200" ]
131131
132+ - name : Cache HuggingFace embedding model
133+ uses : actions/cache@v4
134+ with :
135+ path : /tmp/hf-cache
136+ key : hf-sentence-transformers-all-mpnet-base-v2
137+
138+ - name : Pre-download HuggingFace embedding model
139+ env :
140+ HF_HOME : /tmp/hf-cache
141+ run : |
142+ pip install -q sentence-transformers
143+ python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('sentence-transformers/all-mpnet-base-v2')"
144+ echo "HF_CACHE_PATH=/tmp/hf-cache" >> $GITHUB_ENV
145+
132146 - name : Docker Login for quay access
133147 if : matrix.mode == 'server'
134148 env :
Original file line number Diff line number Diff line change @@ -143,6 +143,20 @@ jobs:
143143 echo "$BODY"
144144 [ "$HTTP_CODE" = "200" ]
145145
146+ - name : Cache HuggingFace embedding model
147+ uses : actions/cache@v4
148+ with :
149+ path : /tmp/hf-cache
150+ key : hf-sentence-transformers-all-mpnet-base-v2
151+
152+ - name : Pre-download HuggingFace embedding model
153+ env :
154+ HF_HOME : /tmp/hf-cache
155+ run : |
156+ pip install -q sentence-transformers
157+ python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('sentence-transformers/all-mpnet-base-v2')"
158+ echo "HF_CACHE_PATH=/tmp/hf-cache" >> $GITHUB_ENV
159+
146160 - name : Docker Login for quay access
147161 env :
148162 QUAY_ROBOT_USERNAME : ${{ secrets.QUAY_DOWNSTREAM_USERNAME }}
You can’t perform that action at this time.
0 commit comments