Update pinned torch version

danbraunai-goodfire · danbraunai-goodfire · commit 7407ff3b205b · 2025-08-13T14:18:36.000Z
diff --git a/pyproject.toml b/pyproject.toml
@@ -5,7 +5,7 @@ description = "Training of small LM models for SimpleStories"
 requires-python = ">=3.11"
 readme = "README.md"
 dependencies = [
-    "torch<2.6.0",
+    "torch>=2.6.0",
     "torchvision",
     "pydantic",
     "wandb",
