Relax torch constraints

Lightning-AI · Dec 28, 2024 · 501846e · 501846e
1 parent fabf765
commit 501846e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -9,9 +9,9 @@ readme = "README.md"
 license = { file = "LICENSE" }
 
 dependencies = [
-    "torch>=2.2.0,<=2.4.1",
+    "torch>=2.5.0,<2.6.0",
     "numpy<2.0",
-    "lightning==2.4.0",
+    "lightning>=2.5.0,<2.6.0",
     "jsonargparse[signatures]>=4.30.1,<=4.32.1",    # 4.33 does not seem to be compatible with Python 3.9
     "huggingface_hub>=0.23.5",          # download models
     "safetensors>=0.4.3",               # download models