Skip to content

Commit c4903d6

Browse files
committed
[WIP] Use pytorch nightly channel w/ stable pytorch b/s why not
1 parent ffdaa52 commit c4903d6

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

.github/workflows/build_vllm.yaml

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ on:
55
branches:
66
- nightly
77
workflow_dispatch:
8+
pull_request: # Remove this before landing
89

910
permissions:
1011
id-token: write
@@ -25,6 +26,7 @@ jobs:
2526
wheel-nightly-policy: gha_workflow_preview_build_wheels
2627
wheel-upload-path: whl/preview/forge/
2728
package-name: forge
29+
channel: test # Hack here to make sure stable pytorch is used
2830
build-matrix: |
2931
{
3032
"include": [
@@ -35,12 +37,12 @@ jobs:
3537
"desired_cuda": "cu128",
3638
"container_image": "pytorch/manylinux2_28-builder:cuda12.8",
3739
"package_type": "manywheel",
38-
"build_name": "manywheel-py3_10-cuda12_8",
40+
"build_name": "manywheel-cuda12_8",
3941
"validation_runner": "linux.12xlarge.memory",
4042
"installation": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128",
4143
"channel": "nightly",
4244
"upload_to_base_bucket": "no",
43-
"stable_version": "2.8.0",
45+
"stable_version": "2.9.0",
4446
"use_split_build": false
4547
}
4648
]

0 commit comments

Comments
 (0)