Diff Coverage

            # The final dtype is applied inside the atomic build (05 stage-5
            # item 5); the Trainer no longer patches it afterwards.
            model_init_dtype=self.config.model_init_dtype,
        )
        _validate_optimize_dtype(
            self.model,
            fp32_main_params=self.config.optimizer.fp32_main_params,
        )
        self.model_config = self.model.config