From 78fd7f6aa8d8317317792e9c71aa5fc89440b008 Mon Sep 17 00:00:00 2001
From: Manu Kashyap <manukashyap20@gmail.com>
Date: Wed, 29 Nov 2023 20:45:03 +0530
Subject: [PATCH] Fixed naming for sentence-transformers library (#4764)

---
 docker/Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index 9feea6e9..2161fb18 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -45,7 +45,7 @@ RUN --mount=type=cache,target=/root/.cache/pip,rw \
     . /app/venv/bin/activate && \
     pip3 install --upgrade pip setuptools wheel ninja && \
     pip3 install torch xformers --index-url https://download.pytorch.org/whl/cu121 && \
-    pip3 install torchvision torchaudio sentence_transformers
+    pip3 install torchvision torchaudio sentence-transformers
 
 # Copy and install GPTQ-for-LLaMa
 COPY --from=builder /build /app/repositories/GPTQ-for-LLaMa