flax

22146233 - Fix automasking: hidden size was too small