Comfy-Org / Comfy-Org/ComfyUI

How to disable Sage attention ?!

Open
#14,860 7 comments 0 reactions 0 assignees View on GitHub
Potential Bug
Dominant language
Python
Stars
133k
Forks
15.7k
Avg merge
1d 7h
Merged PRs (30d)
158

Description

### Custom Node Testing

- [x] I have tried disabling custom nodes and the issue persists (see [how to disable custom nodes](https://docs.comfy.org/troubleshooting/custom-node-issues#step-1%3A-test-with-all-custom-nodes-disabled) if you need help)

### Expected Behavior

removing the --use-sage-attention argument from the bat file should prevent torch from using sage attention which is not the case

### Actual Behavior

Now sage attention auto enables no matter what you do, even if you remove the --use-sage-attention from the command, it's really bad because a lot of models dont play well with sage attention and i find myself everytime having to either modify the files or uninstall the thing completely, i also tried the kijnodes thing with the cuda selected and still nothing, black image on krea2 unless i remove sage completely. And i can't for the life of me find how to disable it with just an argument....

Thanks

### Steps to Reproduce

remove the --use-sage-attention from the bat file and watch it telling you that it is auto enabling sage attention...

### Debug Logs

```powershell
no logs
```

### Other

_No response_

Contributor guide

Open the contributing guide

Research direction

Start with the Windows .bat launch command and search the argument-handling entry point for --use-sage-attention and any automatic Sage attention selection. Reproduce the issue with the argument removed, then verify that Sage attention is not selected and that affected models no longer produce black images.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
ai, machine-learning
Issue type
Bug
Difficulty
3/5
Estimated time
1-2 days
Activity status
Quiet
Clarity
Mostly clear
Newbie friendliness
48/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.