diff --git a/use-cases/model-fine-tuning-pipeline/fine-tuning/pytorch/src/requirements.txt b/use-cases/model-fine-tuning-pipeline/fine-tuning/pytorch/src/requirements.txt index b16348bf..f7368de9 100644 --- a/use-cases/model-fine-tuning-pipeline/fine-tuning/pytorch/src/requirements.txt +++ b/use-cases/model-fine-tuning-pipeline/fine-tuning/pytorch/src/requirements.txt @@ -3,7 +3,7 @@ bitsandbytes==0.44.1 datasets==3.1.0 einops==0.8.0 gcsfs==2024.9.0 -mlflow==2.17.2 +mlflow==2.20.3 peft==v0.13.2 psutil==6.1.0 pynvml==11.5.3