diff --git a/workshops/fine-tuning-with-sagemakerai-and-bedrock/task_01_foundation_model_playground/scripts/requirements.txt b/workshops/fine-tuning-with-sagemakerai-and-bedrock/task_01_foundation_model_playground/scripts/requirements.txt
index 0220b35..5a563fe 100644
--- a/workshops/fine-tuning-with-sagemakerai-and-bedrock/task_01_foundation_model_playground/scripts/requirements.txt
+++ b/workshops/fine-tuning-with-sagemakerai-and-bedrock/task_01_foundation_model_playground/scripts/requirements.txt
@@ -5,7 +5,7 @@ bitsandbytes==0.45.1
 datasets==3.2.0
 evaluate==0.4.3
 huggingface_hub[hf_transfer]==0.33.4
-mlflow==2.22.2
+mlflow==3.5.0rc0
 safetensors>=0.4.5
 sagemaker==2.252.0
 sagemaker-mlflow==0.1.0