fix: replace sentence-transformers with fastembed to eliminate torch/CUDA dependency

2026-06-04 09:40:00 +07:00
parent 1e30885e1a
commit e8e7f02aff
4 changed files with 7 additions and 7 deletions
@@ -11,7 +11,7 @@ COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt

 # Pre-download embedding model agar tidak diunduh ulang saat runtime
-RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('all-MiniLM-L6-v2')"
+RUN python -c "from fastembed import TextEmbedding; TextEmbedding('sentence-transformers/all-MiniLM-L6-v2')"

 COPY src/ ./src/