From b5a103ae1614c2cd9b96a15bb1ab1887d5eedd36 Mon Sep 17 00:00:00 2001 From: Wei-Sheng Chin Date: Sat, 7 Oct 2023 09:51:24 -0700 Subject: [PATCH] Upgrade transformers to fix CI (#17823) Python package pipeline fails due to "tokenizers" compilation. Since "tokenizers" is a dep of "transformers", we update its version and hope a new solution had been there. ``` error: casting `&T` to `&mut T` is undefined behavior, even if the reference is unused, consider instead using an `UnsafeCell` --> tokenizers-lib/src/models/bpe/trainer.rs:517:47 ``` --- .../docker/scripts/training/ortmodule/stage2/requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt index 202d43befcca4..891291b6fa733 100644 --- a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt +++ b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt @@ -2,7 +2,7 @@ pandas scikit-learn numpy==1.21.6 ; python_version < '3.11' numpy==1.24.2 ; python_version >= '3.11' -transformers==v4.4.2 +transformers==v4.16.1 rsa==4.9 tensorboard>=2.2.0,<2.5.0 h5py