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

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

sayakpaul merged 7 commits into

huggingface:mainhuggingface/diffusers:mainfrom

AstraliteHeart:fix-pe_selection_index_based_on_dimAstraliteHeart/diffusers:fix-pe_selection_index_based_on_dimCopy head branch name to clipboard