flax

85dfad24 - Merge pull request #3571 from giovannic:bug/attention_promotion