diffusers
Rewrite AuraFlowPatchEmbed.pe_selection_index_based_on_dim to be torch.compile compatible
#11297

Merged

Rewrite AuraFlowPatchEmbed.pe_selection_index_based_on_dim to be torch.compile compatible #11297

sayakpaul merged 7 commits into huggingface:main from AstraliteHeart:fix-pe_selection_index_based_on_dim

Update pe_selection_index_based_on_dim

85ea323f

Merge branch 'huggingface:main' into af-compile-fix

6bd6b7c3

Make pe_selection_index_based_on_dim work with torh.compile

5c6c6791

sayakpaul approved these changes on 2025-04-12

Merge branch 'main' into fix-pe_selection_index_based_on_dim

a2574c52

sayakpaul requested a review from

yiyixuxu 356 days ago

yiyixuxu approved these changes on 2025-04-14

bobrenjc93 approved these changes on 2025-04-15

Merge branch 'main' into fix-pe_selection_index_based_on_dim

2ce91662

sayakpaul added torch.compile

sayakpaul added Good Example PR

Fix AuraFlowTransformer2DModel's dpcstring default values

ff5674c5

Merge branch 'main' into fix-pe_selection_index_based_on_dim

9bf2facf

sayakpaul merged b6156aaf into main 355 days ago

Reviewers

yiyixuxu

sayakpaul

bobrenjc93

Assignees

No one assigned

Labels

Good Example PR torch.compile

Milestone

No milestone