diff --git a/pyproject.toml b/pyproject.toml
index 48eb38f..4c9dbb7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -13,7 +13,7 @@ classifiers = [
 ]
 requires-python = ">=3.10"
 dependencies = [
-    "torch >= 2.6",
+    "torch >= 2.6, <= 2.8",
     "torchvision",
     "numpy",
     "safetensors",