Update comment on Models

Commit

4 years ago

Update comment on Models Flax Modules don't carry state, but according to this comment they do. Probably a left-over from pre-Linen.