diff --git a/poetry.lock b/poetry.lock
index e2f485c..1121cdf 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -984,13 +984,13 @@ files = [
 
 [[package]]
 name = "mlflow"
-version = "2.5.0"
+version = "2.6.0"
 description = "MLflow: A Platform for ML Development and Productionization"
 optional = false
 python-versions = ">=3.8"
 files = [
-    {file = "mlflow-2.5.0-py3-none-any.whl", hash = "sha256:981fcb3480ca7383b47e22b5e4c726d21e2c87fb4035e5a1b57574736c665576"},
-    {file = "mlflow-2.5.0.tar.gz", hash = "sha256:f992ae8ea9c73502344baf48c4ec447aa9efbfa8965bc090868e6163234f4eb0"},
+    {file = "mlflow-2.6.0-py3-none-any.whl", hash = "sha256:3fc90da77be154f13ad306d94efc6ac77d504107f9085e1262c8c3e259521e79"},
+    {file = "mlflow-2.6.0.tar.gz", hash = "sha256:fd569232e65d69c0cb7006847b1a1bd80831bf0e19378052a44ee5c0ae349182"},
 ]
 
 [package.dependencies]
@@ -1002,7 +1002,7 @@ docker = ">=4.0.0,<7"
 entrypoints = "<1"
 Flask = "<3"
 gitpython = ">=2.1.0,<4"
-gunicorn = {version = "<21", markers = "platform_system != \"Windows\""}
+gunicorn = {version = "<22", markers = "platform_system != \"Windows\""}
 importlib-metadata = ">=3.7.0,<4.7.0 || >4.7.0,<7"
 Jinja2 = [
     {version = ">=2.11,<4", markers = "platform_system != \"Windows\""},
@@ -2170,4 +2170,4 @@ testing = ["big-O", "flake8 (<5)", "jaraco.functools", "jaraco.itertools", "more
 [metadata]
 lock-version = "2.0"
 python-versions = "^3.9"
-content-hash = "00bb11f5e33edcf8495a155bf6a9a81570ac0c1a3cf5806cd52e36d8c5c031dc"
+content-hash = "cf02f1ddf7071c83d0fd0fdc41a2c749e93f4b617333e4ed3d61ce54ff421874"
diff --git a/pyproject.toml b/pyproject.toml
index 0525109..fc443e1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -9,7 +9,7 @@ python = "^3.9"
 torch = "^1.13.1"
 torchvision = "^0.14.1"
 omegaconf = "^2.2.3"
-mlflow = "^2.5.0"
+mlflow = "^2.6.0"
 tqdm = "^4.64.1"
 loguru = "^0.6.0"
 torchmetrics = "^0.10.1"