From fd88474077cdae0653f478b0c7270d14164690b6 Mon Sep 17 00:00:00 2001 From: Jian Chen Date: Thu, 29 Aug 2024 14:53:15 -0700 Subject: [PATCH] Fix a CG issue that require upgrade transformer from 4.36 to 4.38 (#21900) ### Description Fix a CG issue that require upgrade transformer from 4.36 to 4.38 ### Motivation and Context See CG [link](https://aiinfra.visualstudio.com/Lotus/_componentGovernance/218239/alert/11474680?typeId=26218094&pipelinesTrackingFilter=0) Also the other [CG item](https://aiinfra.visualstudio.com/Lotus/_componentGovernance/218239/alert/11474678?typeId=26218094&pipelinesTrackingFilter=0) to request update 4.72 to 4.38 --- cgmanifests/cgmanifest.json | 2 +- .../python/tools/transformers/models/llama/requirements.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/cgmanifests/cgmanifest.json b/cgmanifests/cgmanifest.json index cf245e63a3..1432193ac9 100644 --- a/cgmanifests/cgmanifest.json +++ b/cgmanifests/cgmanifest.json @@ -469,7 +469,7 @@ "type": "pip", "pip": { "Name": "transformers", - "Version": "4.36.0" + "Version": "4.38.0" }, "comments": "Installed in the training docker image" } diff --git a/onnxruntime/python/tools/transformers/models/llama/requirements.txt b/onnxruntime/python/tools/transformers/models/llama/requirements.txt index 388025165f..3ea6915d97 100644 --- a/onnxruntime/python/tools/transformers/models/llama/requirements.txt +++ b/onnxruntime/python/tools/transformers/models/llama/requirements.txt @@ -1,5 +1,5 @@ optimum>=1.14.1 -transformers>=4.33.2,<= 4.37.2 +transformers>=4.33.2,<= 4.38.0 torch>=2.2.0 onnx==1.16.1 datasets>=2.8.0