Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Core] feat: enable fused attention projections for other SD and SDXL pipelines #6179

Merged
merged 3 commits into from
Dec 16, 2023

Conversation

sayakpaul
Copy link
Member

@sayakpaul sayakpaul commented Dec 15, 2023

What does this PR do?

A follow-up of #6030.

Allow the main SD and SDXL pipelines to benefit from horizontal fusion of attention projection matrices.

Related to: https://github.com/huggingface/sdxl-fast.

@kadirnar
Copy link
Contributor

Will @sayakpaul add it for Controlnet? Are there benchmark results?

@@ -23,6 +23,7 @@
from ...image_processor import PipelineImageInput, VaeImageProcessor
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The pipeline is deprecated let's not change it anymore ideally

@HuggingFaceDocBuilderDev

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

@sayakpaul sayakpaul merged commit 2d94c78 into main Dec 16, 2023
16 checks passed
@sayakpaul sayakpaul deleted the enable-fused-projections branch December 16, 2023 03:16
@kadirnar
Copy link
Contributor

@sayakpaul , @patrickvonplaten

Install:

pip install -e.

Usage:
image

I tested this optimization code. Only 3 seconds faster for 10 samples. Is this true?

donhardman pushed a commit to donhardman/diffusers that referenced this pull request Dec 18, 2023
… pipelines (huggingface#6179)

* feat: enable fused attention projections for other SD and SDXL pipelines

* add: test for SD fused projections.
yoonseokjin pushed a commit to yoonseokjin/diffusers that referenced this pull request Dec 25, 2023
… pipelines (huggingface#6179)

* feat: enable fused attention projections for other SD and SDXL pipelines

* add: test for SD fused projections.
AmericanPresidentJimmyCarter pushed a commit to AmericanPresidentJimmyCarter/diffusers that referenced this pull request Apr 26, 2024
… pipelines (huggingface#6179)

* feat: enable fused attention projections for other SD and SDXL pipelines

* add: test for SD fused projections.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants