Skip to content

Commit da5c3f4

Browse files
committed
feat: store embedding cache in volume
1 parent 24cf6b3 commit da5c3f4

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ DOCKER_BUILDKIT=1 docker build . -t gbnc
1212
docker run \
1313
--env HUGGING_FACE_HUB_TOKEN=$HUGGING_FACE_HUB_TOKEN \
1414
--volume "$(pwd)/gswikichat":/workspace/gswikichat \
15-
--volume "$(pwd)/cache":/root/.cache \
15+
--volume gbnc_cache:/root/.cache
1616
--publish 8000:8000 \
1717
--rm \
1818
--interactive \

gswikichat/vector_store_interface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
HUGGING_FACE_HUB_TOKEN = os.environ.get('HUGGING_FACE_HUB_TOKEN')
2222

2323
# disable this line to disable the embedding cache
24-
EMBEDDING_CACHE_FILE = '/tmp/gbnc_embeddings.json'
24+
EMBEDDING_CACHE_FILE = '/root/.cache/gbnc_embeddings.json'
2525

2626
top_k = 5
2727
input_documents = []

0 commit comments

Comments
 (0)