diff --git a/Container-Root/hyperpod/deployment/eks/demo/hero/training/fsdp/src/requirements.txt b/Container-Root/hyperpod/deployment/eks/demo/hero/training/fsdp/src/requirements.txt
index f896d6c..185b161 100644
--- a/Container-Root/hyperpod/deployment/eks/demo/hero/training/fsdp/src/requirements.txt
+++ b/Container-Root/hyperpod/deployment/eks/demo/hero/training/fsdp/src/requirements.txt
@@ -4,5 +4,5 @@ torch==2.7.1
 torchaudio==2.7.1
 torchvision==0.22.1
 transformers==4.52.4
-mlflow==3.0.0
+mlflow==3.5.0rc0
 sagemaker-mlflow
\ No newline at end of file