diff --git a/Dockerfile b/Dockerfile index 366a8db67c25c41634af8afb86995cc126415052..a10345763708e5ba01bf19261c500a2115208eca 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,11 +1,9 @@ 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 \ - hdbscan umap-learn \ + sentence-transformers 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}"