flax

32ad6c57 - Rollback of attention_bias parameter, because the change overrides the attention bias for injected attention functions.