update dependency

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

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

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 && \

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'
 ]