diff --git a/assets/training/model_evaluation/environment/context/Dockerfile b/assets/training/model_evaluation/environment/context/Dockerfile
index eb694c324f..daff14b994 100644
--- a/assets/training/model_evaluation/environment/context/Dockerfile
+++ b/assets/training/model_evaluation/environment/context/Dockerfile
@@ -25,3 +25,4 @@ ENV LD_LIBRARY_PATH $CONDA_PREFIX/lib:$LD_LIBRARY_PATH
 
 # Temp fix for env vuln
 RUN pip install mlflow=={{latest-pypi-version}}
+RUN pip install transformers=={{latest-pypi-version}}
diff --git a/assets/training/model_evaluation/environment/context/conda.yaml b/assets/training/model_evaluation/environment/context/conda.yaml
index 804a16e9ac..ec255d5250 100644
--- a/assets/training/model_evaluation/environment/context/conda.yaml
+++ b/assets/training/model_evaluation/environment/context/conda.yaml
@@ -11,7 +11,7 @@ dependencies:
 - pip:
   - aiohttp>=3.10.2
   - matplotlib>=3.7,<3.8
-  - protobuf<4.0.0
+  - protobuf<7.0.0
   - psutil>=5.8,<5.9
   - pandas>=2.0.2,<2.1.0
   - azureml-core=={{latest-pypi-version}}