stable monarch

Author: pbontrager

pyproject.toml

@@ -3,7 +3,7 @@
 name = "forge"
 description = "A PyTorch native platform for post-training generative AI models"
 readme = "README.md"
-requires-python = ">=3.10, <=3.13"
+requires-python = ">=3.10, <3.13"
 license = {file = "LICENSE"}
 authors = [
     { name = "PyTorch Team", email = "[email protected]" },
@@ -13,10 +13,10 @@ dependencies = [
     # PyTorch
     "torch==2.9.0",
     "torchdata>=0.8.0",
-    "torchtitan",
-    "torchmonarch==0.1.0rc4",
-    "torchstore==0.0.1rc2",
-    "vllm==0.11.0",
+    "torchtitan==0.2.0",
+    "torchmonarch==0.1.2",
+    "torchstore==0.1.2",
+    "vllm",
     # Hugging Face integrations
     "datasets>=2.21.0",
     "tokenizers",
@@ -63,16 +63,19 @@ members = [
     "forge",
 ]
 
-# pytorch
+# [[tool.uv.index]]
+# name = "pytorch-nightly"
+# url  = "https://download.pytorch.org/whl/nightly"
+
 [[tool.uv.index]]
-name = "pytorch-nightly"
-url  = "https://download.pytorch.org/whl/nightly"
+name = "forge"
+url = "https://download.pytorch.org/whl/preview/forge/"
 
 [tool.uv.sources]
 # torchtitan = { index = "pytorch-nightly" }
 # torch = { index = "pytorch-nightly" }
 # torchstore = { git = "ssh://[email protected]/meta-pytorch/torchstore.git"}
-# vllm = { index = "pytorch-nightly" }
+vllm = { index = "forge" }
 
 [tool.uv]
 # TODO: revert to stricter default uv strategy
@@ -82,4 +85,4 @@ prerelease = "allow"
 environments = [
     "sys_platform == 'linux'",
 ]
-override-dependencies = ["torch>2.7.1"] # , "torchaudio>=2.7.1", "torchvision>=0.22.0"]
+#override-dependencies = ["torch>2.7.1"] # , "torchaudio>=2.7.1", "torchvision>=0.22.0"]