Tests currently initialize components manually, we should extract components directly from the model (eg model.transformer not DynaFormer()).