diff --git a/assets/wheels/monarch-0.0.1-cp310-cp310-linux_x86_64.whl b/assets/wheels/monarch-0.0.1-cp310-cp310-linux_x86_64.whl index 2e5d7741a..e6901b3f3 100644 Binary files a/assets/wheels/monarch-0.0.1-cp310-cp310-linux_x86_64.whl and b/assets/wheels/monarch-0.0.1-cp310-cp310-linux_x86_64.whl differ diff --git a/assets/wheels/torchstore-0.1.0-py3-none-any.whl b/assets/wheels/torchstore-0.1.0-py3-none-any.whl index ee92071e4..9c7ad8f4f 100644 Binary files a/assets/wheels/torchstore-0.1.0-py3-none-any.whl and b/assets/wheels/torchstore-0.1.0-py3-none-any.whl differ diff --git a/assets/wheels/torchtitan-0.1.0-py3-none-any.whl b/assets/wheels/torchtitan-0.1.0-py3-none-any.whl index 587e533da..1bc1f2c83 100644 Binary files a/assets/wheels/torchtitan-0.1.0-py3-none-any.whl and b/assets/wheels/torchtitan-0.1.0-py3-none-any.whl differ diff --git a/scripts/build_wheels.sh b/scripts/build_wheels.sh index 14f6db362..a36a494b8 100755 --- a/scripts/build_wheels.sh +++ b/scripts/build_wheels.sh @@ -9,11 +9,11 @@ BLUE='\033[0;34m' NC='\033[0m' # Configuration -PYTORCH_VERSION="2.9.0.dev20250828" +PYTORCH_VERSION="2.9.0.dev20250905" VLLM_BRANCH="v0.10.0" -MONARCH_COMMIT="b8a047359462c134bc9f1c48d7d9b8c2248913d2" -TORCHTITAN_COMMIT="298bf48212adf36c60690f0a8be7481737890828" -TORCHSTORE_COMMIT="35430b4be4f26519704b1c7db07deccae87c883b" +MONARCH_COMMIT="ce078ddbbac6b6f3682afd763d8d96dad8092a91" +TORCHTITAN_COMMIT="60645bc78cf4138ae54e8f29a8b24bf0ccd64404" +TORCHSTORE_COMMIT="dae7e5cbc2b23c294e14fd5be86a38411595bed4" BUILD_DIR="$HOME/forge-build" WHEEL_DIR="$(pwd)/assets/wheels" diff --git a/scripts/install.sh b/scripts/install.sh index 036eba40e..b95518320 100755 --- a/scripts/install.sh +++ b/scripts/install.sh @@ -18,10 +18,10 @@ log_error() { echo -e "${RED}[ERROR]${NC} $1"; } log_warning() { echo -e "${YELLOW}[WARNING]${NC} $1";} # Configuration -PYTORCH_VERSION="2.9.0.dev20250828" +PYTORCH_VERSION="2.9.0.dev20250905" SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" WHEEL_DIR="$SCRIPT_DIR/../assets/wheels" -RELEASE_TAG="v0.0.0" +RELEASE_TAG="v0.0.0-92025" GITHUB_REPO="meta-pytorch/forge" # Check conda environment @@ -165,9 +165,16 @@ download_vllm_wheel() { if [ -z "$vllm_wheel_name" ]; then log_error "Could not find vLLM wheel in release $RELEASE_TAG" - log_info "Make sure you've uploaded the vLLM wheel to the GitHub release" + log_info "Make sure the vLLM wheel has been uploaded to the GitHub release" exit 1 fi + for f in assets/wheels/vllm-*; do + [ -e "$f" ] || continue # skip if glob didn't match + if [ "$(basename "$f")" != "$vllm_wheel_name" ]; then + log_info "Removing stale vLLM wheel: $(basename "$f")" + rm -f "$f" + fi + done local local_path="$WHEEL_DIR/$vllm_wheel_name"