A production host can learn the production state awareness (PSA) modes supported by a memory device and select a particular of one of the supported PSA modes. The memory device can receive host image data from the production host and write the host image data according to the selected PSA mode. The memory device can set a PSA state to load complete after writing the host image data. The memory device can thereby be better situated for being soldered to a memory sub-system.