aapot's picture
Add 480k train step and HF flax model
c2701e3