flax
Rollback of attention_bias parameter, because the change overrides the attention bias for injected attention functions.
#4703
Merged
Merged