From 7280bf7e75c1132b1fff11b706164ae45067a9a4 Mon Sep 17 00:00:00 2001 From: zhouyu5 Date: Thu, 26 Oct 2023 15:08:29 +0000 Subject: [PATCH] update dependency --- Dockerfile-ubuntu/Dockerfile-v1.2 | 4 ++-- e2eAIOK/deltatuner/docker/Dockerfile | 4 ++-- e2eAIOK/deltatuner/docker/DockerfileGPU | 2 +- e2eAIOK/deltatuner/setup.py | 2 +- 4 files changed, 6 insertions(+), 6 deletions(-) diff --git a/Dockerfile-ubuntu/Dockerfile-v1.2 b/Dockerfile-ubuntu/Dockerfile-v1.2 index 4fede3352..1a3164e35 100644 --- a/Dockerfile-ubuntu/Dockerfile-v1.2 +++ b/Dockerfile-ubuntu/Dockerfile-v1.2 @@ -12,9 +12,9 @@ RUN apt-get -y update \ && apt-get clean \ && rm -rf /var/lib/apt/lists/* -RUN pip install torch==1.13.0 transformers datasets accelerate SentencePiece \ +RUN pip install torch>=1.13.1 transformers datasets SentencePiece \ evaluate peft==0.4.0 torchsummary \ - nltk rouge_score protobuf==3.20.1 tokenizers einops wandb \ + nltk rouge_score einops wandb \ sigopt RUN pip install --upgrade pip diff --git a/e2eAIOK/deltatuner/docker/Dockerfile b/e2eAIOK/deltatuner/docker/Dockerfile index 59ae92cc9..e1ded7e96 100644 --- a/e2eAIOK/deltatuner/docker/Dockerfile +++ b/e2eAIOK/deltatuner/docker/Dockerfile @@ -52,7 +52,7 @@ RUN conda init bash && \ conda create -yn chatbot-finetuning python=3.9 && \ echo "conda activate chatbot-finetuning" >> ~/.bashrc && \ source ~/.bashrc && \ - pip install torch==1.13.0 transformers datasets accelerate SentencePiece \ + pip install torch>=1.13.1 transformers datasets SentencePiece \ evaluate peft==0.4.0 torchsummary \ - nltk rouge_score protobuf==3.20.1 tokenizers einops wandb \ + nltk rouge_score einops wandb \ sigopt diff --git a/e2eAIOK/deltatuner/docker/DockerfileGPU b/e2eAIOK/deltatuner/docker/DockerfileGPU index 60cab9697..18d0d4f27 100644 --- a/e2eAIOK/deltatuner/docker/DockerfileGPU +++ b/e2eAIOK/deltatuner/docker/DockerfileGPU @@ -40,7 +40,7 @@ RUN conda init bash && \ source ~/.bashrc && \ pip install transformers datasets accelerate SentencePiece \ peft==0.4.0 torchsummary sigopt evaluate \ - nltk rouge_score protobuf==3.20.1 tokenizers einops \ + nltk rouge_score tokenizers einops \ wandb RUN git clone https://github.com/EleutherAI/lm-evaluation-harness.git ~/lm-evaluation-harness && \ diff --git a/e2eAIOK/deltatuner/setup.py b/e2eAIOK/deltatuner/setup.py index 059ab99df..289cc6eb2 100644 --- a/e2eAIOK/deltatuner/setup.py +++ b/e2eAIOK/deltatuner/setup.py @@ -7,7 +7,7 @@ VERSION = fh.read().strip() REQUIRED_PACKAGES = [ - 'torch>=1.13.0', 'transformers', 'datasets', 'sentencepiece', + 'torch>=1.13.1', 'transformers', 'datasets', 'sentencepiece', 'peft==0.4.0', 'evaluate', 'nltk', 'rouge_score', 'einops', 'sigopt', 'torchsummary' ]