Use Monarch 0.1.0rc3 & PyTorch 2.9.0 stable

Author: joecummings

assets/versions.sh

@@ -7,13 +7,9 @@
 # Version Configuration for Forge Wheel Building
 # This file contains all pinned versions and commits for dependencies
 
-# PyTorch version
-PYTORCH_VERSION="2.9.0.dev20250905"
+PYTORCH_VERSION="2.9.0"
+MONARCH_VERSION="0.1.0rc3"
 
-# vLLM branch
 VLLM_BRANCH="v0.10.0"
-
-# Commit hashes
-MONARCH_COMMIT="195503223b5c2896846171f60ac99dc6868f8f2c"
 TORCHTITAN_COMMIT="0cfbd0b3c2d827af629a107a77a9e47229c31663"
 TORCHSTORE_COMMIT="662299faf4fd50ee30bd9aa3f4ce8c0e2db1d310"

scripts/install.sh

@@ -274,21 +274,24 @@ main() {
     check_gh_install
     download_vllm_wheel
 
-    log_info "Installing PyTorch nightly..."
-    pip install torch==$PYTORCH_VERSION --index-url https://download.pytorch.org/whl/nightly/cu129
+    log_info "Installing PyTorch..."
+    if [[ "$PYTORCH_VERSION" == *"dev"* ]]; then
+        pip install --pre torch==$PYTORCH_VERSION --index-url https://download.pytorch.org/whl/nightly/cu128
+    else
+        pip install torch==$PYTORCH_VERSION
+    fi
+
+    log_info "Installing Monarch ..."
+    pip install torchmonarch==$MONARCH_VERSION
 
-    log_info "Installing all wheels (local + downloaded)..."
+    log_info "Installing remaining wheels (local + downloaded)..."
     pip install "$WHEEL_DIR"/*.whl
 
     log_info "Installing Forge from source..."
     pip install -e .
 
-    # Set up environment
     log_info "Setting up environment..."
-
-     # Get conda environment directory
     local conda_env_dir="${CONDA_PREFIX}"
-
     if [ -z "$conda_env_dir" ]; then
         log_error "Could not determine conda environment directory"
         exit 1