diff --git a/Dockerfile b/Dockerfile
index a10345763708e5ba01bf19261c500a2115208eca..366a8db67c25c41634af8afb86995cc126415052 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,9 +1,11 @@
 FROM quay.io/jupyter/pytorch-notebook:cuda12-pytorch-2.3.1
 
+# Install Python packages that may be useful.
 RUN pip install --no-cache-dir \
         transformers sentence-transformers accelerate bitsandbytes \
-        sentence-transformers hdbscan umap-learn \
+        hdbscan umap-learn \
     && \
+    pip install llama-cpp-python --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu121 && \
     fix-permissions "${CONDA_DIR}" && \
     fix-permissions "/home/${NB_USER}"