code: def format_state_key(cls, key):
      return key.replace("detr_model.", "unit_base_model.")
  
imports: import logging
  from copy import deepcopy
  from typing import Dict
  import torch
  from mmf.common.registry import registry
  from mmf.models import BaseModel
  from mmf.models.unit.unit_base_model import AttributeHead, build_detection_loss, MLP, UniTBaseModel
  from mmf.modules.encoders import TransformerEncoder
  from mmf.utils.distributed import byte_tensor_to_object
  from torch import nn, Tensor
label_desc: Use image node_id 3 for calling the UniT obj's underlying member method code with example usage: obj.format_state_key(cls, key) and returns: key