update datasets version for docker (intel#1077)

mengfei25 · Dec 27, 2023 · a32e184 · a32e184
1 parent b1b407f
commit a32e184
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/docker/Dockerfile_chatbot b/docker/Dockerfile_chatbot
@@ -124,6 +124,7 @@ RUN cd /intel-extension-for-transformers/intel_extension_for_transformers/neural
     pip install -r requirements_hpu.txt && \
     pip install transformers==4.34.1 && \
     pip install accelerate==0.24.0 && \
+    pip install datasets==2.14.7 && \
     pip install pymysql
 
 WORKDIR /intel-extension-for-transformers/intel_extension_for_transformers/neural_chat/

diff --git a/examples/huggingface/pytorch/language-modeling/nas/README.md b/examples/huggingface/pytorch/language-modeling/nas/README.md
@@ -10,6 +10,7 @@ Recommend python 3.9 or higher version.
 ```shell
 pip install -r requirements.txt
 pip install transformers==4.34.1
+pip install datasets==2.14.7
 ```
 >**Note**: Please use transformers no higher than 4.34.1
 

diff --git a/intel_extension_for_transformers/neural_chat/docker/Dockerfile b/intel_extension_for_transformers/neural_chat/docker/Dockerfile
@@ -128,6 +128,7 @@ RUN cd /intel-extension-for-transformers/intel_extension_for_transformers/neural
     pip install -r requirements_hpu.txt && \
     pip install transformers==4.34.1 && \
     pip install accelerate==0.24.0 && \
+    pip install datasets==2.14.7 && \
     pip install pymysql && \
     pip uninstall -y intel_extension_for_pytorch