def METHOD_NAME(self, state_dict: dict) -> None:
    # TODO(ver217): design a new method to load state_dict. When using NVMe offload, whole state_dict may not be able to fit in memory.
    if self.offloader is not None:
        raise NotImplementedError
    super().METHOD_NAME(state_dict)