* Fix init and repro * comment + black --------- Co-authored-by: Srini Iyer <sviyer@meta.com>
Summary: - Refactor local model configs to be separate and clearer - Add attention arguments and correct which attention is used in local models - Preparation for being able to have an entropy train script - Fix failing unit tests Test Plan: