Use native memory efficient attention in PyTorch 2.0 if possible#2778
Use native memory efficient attention in PyTorch 2.0 if possible#2778haotian-liu wants to merge 1 commit intohuggingface:mainfrom
Conversation
|
The documentation is not available anymore as the PR was closed or merged. |
|
@haotian-liu this class is to be deprecated and in fact, the PR to remove it is #2697 The https://github.com/huggingface/diffusers/blob/main/src/diffusers/models/attention_processor.py#L109 |
|
This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread. Please note that issues that do not follow the contributing guidelines are likely to be ignored. |
|
Hey @haotian-liu , I think this has been fixed with: #3200 |
|
Great, thank you! Closing this as fixed in #3200. |
When users use PyTorch 2.0, and do not explicit enable memory efficient attention with xformers, this can potentially lead to OOM issues (while the user may believe that the efficient attention is automatically enabled with PyTorch 2.0).