From 8a5603e5b0d9b8a90fcc1d6a73a7475c97bd6b43 Mon Sep 17 00:00:00 2001 From: sfc-gh-zhwang Date: Mon, 2 Oct 2023 15:19:13 -0700 Subject: [PATCH 1/2] commit --- docker/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index bfbfba0..11c38c5 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -41,7 +41,7 @@ RUN apt-get update && \ RUN pip3 install --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cu118 torch==2.0.1+cu118 && \ pip3 install --no-cache-dir --extra-index-url https://pypi.ngc.nvidia.com regex fire tritonclient[all] && \ - pip3 install --no-cache-dir accelerate transformers huggingface_hub tokenizers SentencePiece sacrebleu datasets tqdm omegaconf rouge_score && \ + pip3 install --no-cache-dir tokenizers sacrebleu datasets tqdm omegaconf rouge_score && \ pip3 install --no-cache-dir cmake==3.24.3 # backend build From b5e0c4a7fa16e577d38a6fd856245d8827eead7e Mon Sep 17 00:00:00 2001 From: sfc-gh-zhwang Date: Mon, 2 Oct 2023 15:20:31 -0700 Subject: [PATCH 2/2] commit --- docker/Dockerfile | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 11c38c5..c837402 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -39,8 +39,7 @@ RUN apt-get update && \ apt-get clean && \ rm -rf /var/lib/apt/lists/* -RUN pip3 install --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cu118 torch==2.0.1+cu118 && \ - pip3 install --no-cache-dir --extra-index-url https://pypi.ngc.nvidia.com regex fire tritonclient[all] && \ +RUN pip3 install --no-cache-dir --extra-index-url https://pypi.ngc.nvidia.com regex fire tritonclient[all] && \ pip3 install --no-cache-dir tokenizers sacrebleu datasets tqdm omegaconf rouge_score && \ pip3 install --no-cache-dir cmake==3.24.3