diff --git a/pyproject.toml b/pyproject.toml
index 1d9f720..24a7d0d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -38,7 +38,7 @@ classifiers = [
 
 [tool.poetry.dependencies]
 python = ">=3.10,<4.0"
-torch = "2.11.0"
+torch = ">=2.1.0"
 transformers = ">=4.40.0"
 datasets = ">=2.18.0"
 
diff --git a/training/requirements.txt b/training/requirements.txt
index e3348c5..1461dcd 100644
--- a/training/requirements.txt
+++ b/training/requirements.txt
@@ -1,4 +1,4 @@
-torch>=2.11.0
+torch>=2.1.0
 datasets>=3.6.0
 loguru>=0.7.3
 open-mythos
\ No newline at end of file