Skip to content

Commit b00c7ab

Browse files
committed
fix: update requirements for CPU-only local embeddings
1 parent 9d738ac commit b00c7ab

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)