Skip to content

Commit 45b2641

Browse files
authored
Stable pytorch w/ vllm (#476)
1 parent ff3290e commit 45b2641

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

.github/workflows/build_vllm.yaml

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ jobs:
2525
wheel-nightly-policy: gha_workflow_preview_build_wheels
2626
wheel-upload-path: whl/preview/forge/
2727
package-name: forge
28+
channel: test # Hack here to make sure stable pytorch is used
2829
build-matrix: |
2930
{
3031
"include": [
@@ -35,12 +36,12 @@ jobs:
3536
"desired_cuda": "cu128",
3637
"container_image": "pytorch/manylinux2_28-builder:cuda12.8",
3738
"package_type": "manywheel",
38-
"build_name": "manywheel-py3_10-cuda12_8",
39+
"build_name": "manywheel-cuda12_8",
3940
"validation_runner": "linux.12xlarge.memory",
4041
"installation": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128",
4142
"channel": "nightly",
4243
"upload_to_base_bucket": "no",
43-
"stable_version": "2.8.0",
44+
"stable_version": "2.9.0",
4445
"use_split_build": false
4546
}
4647
]

0 commit comments

Comments
 (0)