Use independent rope method from attention_flax in flux transformer#176

entrpn merged 1 commit intomainAI-Hypercomputer/maxdiffusion:mainfrom

kunjanp-dev-2AI-Hypercomputer/maxdiffusion:kunjanp-dev-2Copy head branch name to clipboard