saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-14 20:48:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

pin transformers dependence to sentencepiece==0.1.92 due to ci fail (#5607)

Browse source
This commit is contained in:
Suffian Khan 2020-10-27 16:21:40 -07:00 committed by GitHub
parent ddf83d1ace
commit e5b0d192f4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
dockerfiles/Dockerfile.training
View file

@ -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