From 28998c5c778dbe91125959acb3a9d5f2459543d6 Mon Sep 17 00:00:00 2001
From: jeff-shepherd <39775772+jeff-shepherd@users.noreply.github.com>
Date: Thu, 27 Feb 2025 11:38:42 -0800
Subject: [PATCH] Upgrade optimum to allow transformers=4.48.0 (#3882)

---
 .../ai-ml-automl-dnn-text-gpu-ptca/context/Dockerfile           | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/assets/training/automl/environments/ai-ml-automl-dnn-text-gpu-ptca/context/Dockerfile b/assets/training/automl/environments/ai-ml-automl-dnn-text-gpu-ptca/context/Dockerfile
index 71c46ad505..8c6f25773f 100644
--- a/assets/training/automl/environments/ai-ml-automl-dnn-text-gpu-ptca/context/Dockerfile
+++ b/assets/training/automl/environments/ai-ml-automl-dnn-text-gpu-ptca/context/Dockerfile
@@ -16,7 +16,7 @@ RUN pip install torch-ort==1.18.0 && TORCH_CUDA_ARCH_LIST="5.2;6.0;7.0;8.0;8.6;9
 RUN pip uninstall -y onnxruntime
 
 RUN pip install transformers==4.48.0
-RUN pip install optimum==1.21.4
+RUN pip install optimum==1.23.3
 RUN pip install accelerate==0.33.0
 RUN pip install deepspeed~=0.15.1
 RUN pip install numpy==1.22.0