SIGN IN SIGN UP

Merge pull request #431 from AI-Hypercomputer:custom_attn_fix

PiperOrigin-RevId: 950894040
M
maxdiffusion authors committed
ed9ac9db21a652e8f203d97585c76b5d301d38ff