Skip to content

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

Merged
sayakpaul merged 7 commits intohuggingface:mainfrom
AstraliteHeart:fix-pe_selection_index_based_on_dim
Apr 15, 2025
Merged

Rewrite AuraFlowPatchEmbed.pe_selection_index_based_on_dim to be torch.compile compatible#11297
sayakpaul merged 7 commits intohuggingface:mainfrom
AstraliteHeart:fix-pe_selection_index_based_on_dim

Commits