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 38ea01f..e989362 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,6 +4,6 @@ torch==2.4.0
 torchaudio==2.4.0
 torchvision==0.19.0
 transformers==4.44.0
-mlflow==3.0.0
+mlflow==3.5.0
 sagemaker-mlflow
 psutil
\ No newline at end of file