Skip to content

Commit

Permalink
Add another ROCm tip.
Browse files Browse the repository at this point in the history
  • Loading branch information
comfyanonymous committed Dec 4, 2024
1 parent 4e402b1 commit 3bed56b
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -219,6 +219,8 @@ You can enable experimental memory efficient attention on pytorch 2.5 in ComfyUI

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

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.

# Notes

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

0 comments on commit 3bed56b

Please sign in to comment.