Skip to content

Update examples for Accelerated Transformers/SDPA #2284

Open
@chauhang

Description

@chauhang

🚀 The feature

Upgrade the Transformer and Diffuser model examples to use the accelerated transformers/sdpa feature from PT 2.0

Blogs with details: Diffusers, Transformers

Motivation, pitch

Out of box performance speedup for the examples

Alternatives

No response

Additional context

No response

Metadata

Metadata

Labels

documentationImprovements or additions to documentationexampletriagedIssue has been reviewed and triaged

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions