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 f128466..83c0f52 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==5.0.0rc3 -mlflow==3.11.0rc1 +mlflow==3.11.1 sagemaker-mlflow psutil \ No newline at end of file