Skip to content

Commit e9af97b

Browse files
Use torch cu129 for nvidia pytorch nightly. (Comfy-Org#8786)
* update nightly workflow with cu129 * Remove unused file to lower standalone size.
1 parent d927730 commit e9af97b

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

.github/workflows/windows_release_nightly_pytorch.yml

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ on:
77
description: 'cuda version'
88
required: true
99
type: string
10-
default: "128"
10+
default: "129"
1111

1212
python_minor:
1313
description: 'python minor version'
@@ -19,7 +19,7 @@ on:
1919
description: 'python patch version'
2020
required: true
2121
type: string
22-
default: "2"
22+
default: "5"
2323
# push:
2424
# branches:
2525
# - master
@@ -53,6 +53,8 @@ jobs:
5353
ls ../temp_wheel_dir
5454
./python.exe -s -m pip install --pre ../temp_wheel_dir/*
5555
sed -i '1i../ComfyUI' ./python3${{ inputs.python_minor }}._pth
56+
57+
rm ./Lib/site-packages/torch/lib/dnnl.lib #I don't think this is actually used and I need the space
5658
cd ..
5759
5860
git clone --depth 1 https://github.com/comfyanonymous/taesd

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -243,7 +243,7 @@ Nvidia users should install stable pytorch using this command:
243243

244244
This is the command to install pytorch nightly instead which might have performance improvements.
245245

246-
```pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128```
246+
```pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129```
247247

248248
#### Troubleshooting
249249

0 commit comments

Comments
 (0)