diff --git a/poetry.lock b/poetry.lock
index 198f331..5e0d340 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -2614,4 +2614,4 @@ tensorflow = ["tensorflow"]
 [metadata]
 lock-version = "2.1"
 python-versions = ">=3.9,<3.13"
-content-hash = "b8046d35d482d57b1a1211744ae14cb2a2d798c7625f72bc10aa7e7866b5be34"
+content-hash = "c5efa85f51a4fec35aa6881590714f69bee91a541e9874fc8c7b876d40b76bb6"
diff --git a/pyproject.toml b/pyproject.toml
index 57a3890..f8464cd 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -34,7 +34,7 @@ requests = "^2.31.0"
 aiohttp = "^3.8.5"
 dill = "^0.3.7"
 types-requests = ">1.26"
-torch = "^2.2"
+torch = "^2.5.1"
 tf-keras = "^2.18.0"