From e5b0d192f4f9910d6188f1d3c5d8f6ce3caf37a7 Mon Sep 17 00:00:00 2001 From: Suffian Khan Date: Tue, 27 Oct 2020 16:21:40 -0700 Subject: [PATCH] pin transformers dependence to sentencepiece==0.1.92 due to ci fail (#5607) --- dockerfiles/Dockerfile.training | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dockerfiles/Dockerfile.training b/dockerfiles/Dockerfile.training index 8ddb0d82cb..6444b18e1e 100644 --- a/dockerfiles/Dockerfile.training +++ b/dockerfiles/Dockerfile.training @@ -151,7 +151,7 @@ RUN cd /stage && git clone https://github.com/microsoft/onnxruntime.git &&\ pip install build/${BUILD_CONFIG}/dist/*.whl # Install AzureML support and commonly used packages. -RUN pip install azureml-defaults transformers==2.11.0 msgpack==1.0.0 tensorboardX==1.8 tensorboard==2.3.0 +RUN pip install azureml-defaults sentencepiece==0.1.92 transformers==2.11.0 msgpack==1.0.0 tensorboardX==1.8 tensorboard==2.3.0 # switch to cuda runtime environment # note: launch with --gpus all or nvidia-docker