Skip to content

Commit 92ebfc9

Browse files
authored
Merge pull request #45 from devtron-labs/fix/docker-image-size
fix: update requirements for CPU-only local embeddings
2 parents 9d738ac + b00c7ab commit 92ebfc9

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

devtron-docs-rag-server/requirements.txt

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,10 @@ uvicorn[standard]>=0.27.0
66
psycopg2-binary>=2.9.9
77
pgvector>=0.2.4
88

9-
# Local Embeddings (supports both CPU and GPU)
9+
# Local Embeddings (CPU-only, avoids ~2.4 GB CUDA wheels)
10+
# Primary index is the PyTorch CPU wheel index; PyPI used as fallback for everything else
11+
--index-url https://download.pytorch.org/whl/cpu
12+
--extra-index-url https://pypi.org/simple
1013
sentence-transformers>=2.2.2
1114
torch>=2.0.0
1215

0 commit comments

Comments
 (0)