Skip to content

Commit d10e808

Browse files
committed
fix: Provide MacOS workaround to spin up CPU embeddings
1 parent 1a31259 commit d10e808

File tree

3 files changed

+51
-0
lines changed

3 files changed

+51
-0
lines changed

.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -99,3 +99,4 @@ clients/trieve-shopify-extension/shopify.app.skept-laptop-trieve-extension.toml
9999
charts/charts/**/*.tgz
100100
clients/trieve-shopify-extension/.env.skept-trieve-extension-test
101101
.idea
102+
docker-compose-cpu-embeddings.override.yml

README.md

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -175,6 +175,35 @@ cat .env.chat .env.search .env.server .env.docker-compose > .env
175175
./convenience.sh -l
176176
```
177177

178+
### Start embedding servers
179+
180+
We offer 2 docker-compose files for embedding servers. One for GPU and one for CPU.
181+
182+
```sh
183+
docker compose -f docker-compose-cpu-embeddings.yml up -d
184+
```
185+
186+
or
187+
188+
```sh
189+
docker compose -f docker-compose-gpu-embeddings.yml up -d
190+
```
191+
192+
* Note on embedding servers. If you want to use a separate GPU enabled device for embedding servers you will need to update the following parameters
193+
194+
```
195+
SPARSE_SERVER_QUERY_ORIGIN
196+
SPARSE_SERVER_DOC_ORIGIN
197+
EMBEDDING_SERVER_ORIGIN
198+
SPARSE_SERVER_QUERY_ORIGIN
199+
```
200+
201+
#### Using CPU embeddings on MacOS
202+
```
203+
cp docker-compose-cpu-embeddings.override.mac docker-compose-cpu-embeddings.override.yml
204+
docker-compose -f docker-compose-cpu-embeddings.yml -f docker-compose-cpu-embeddings.override.yml docker up
205+
```
206+
178207
### Install front-end packages for local dev
179208

180209
```
Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
1+
# Docker Compose overrides for macOS development with CPU embeddings
2+
# To use this file: docker-compose -f docker-compose-cpu-embeddings.yml -f docker-compose-cpu-embeddings.override.yml up
3+
# For Linux development, don't include this override file
4+
5+
version: '3'
6+
7+
services:
8+
splade-doc:
9+
platform: linux/amd64
10+
11+
splade-query:
12+
platform: linux/amd64
13+
14+
jina:
15+
platform: linux/amd64
16+
17+
bgem3:
18+
platform: linux/amd64
19+
20+
reranker:
21+
platform: linux/amd64

0 commit comments

Comments
 (0)