diff --git a/python/batch-predictor/requirements.txt b/python/batch-predictor/requirements.txt
index 9a52ee8bc..37d62149a 100644
--- a/python/batch-predictor/requirements.txt
+++ b/python/batch-predictor/requirements.txt
@@ -3,4 +3,5 @@ findspark==2.0.1
 mlflow==2.16.2
 pyarrow>=0.14.1,<=17.0.0
 pyspark==3.0.1
-setuptools<75
\ No newline at end of file
+setuptools<75
+git+https://github.com/caraml-dev/merlin.git@upgrade_mlflow_version#subdirectory=python/sdk
\ No newline at end of file
diff --git a/python/pyfunc-server/requirements.txt b/python/pyfunc-server/requirements.txt
index 290438376..6c4bea249 100644
--- a/python/pyfunc-server/requirements.txt
+++ b/python/pyfunc-server/requirements.txt
@@ -9,4 +9,5 @@ orjson>=2.6.8
 prometheus-client
 setuptools<75
 tornado
-uvloop
\ No newline at end of file
+uvloop
+git+https://github.com/caraml-dev/merlin.git@upgrade_mlflow_version#subdirectory=python/sdk
\ No newline at end of file