Skip to content

Commit 3bed56b

Browse files
Add another ROCm tip.
1 parent 4e402b1 commit 3bed56b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

README.md

+2
Original file line numberDiff line numberDiff line change
@@ -219,6 +219,8 @@ You can enable experimental memory efficient attention on pytorch 2.5 in ComfyUI
219219

220220
```TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1 python main.py --use-pytorch-cross-attention```
221221

222+
You can also try setting this env variable `PYTORCH_TUNABLEOP_ENABLED=1` which might speed things up at the cost of a very slow initial run.
223+
222224
# Notes
223225

224226
Only parts of the graph that have an output with all the correct inputs will be executed.

0 commit comments

Comments
 (0)