code: def build_container_map(self) -> ContainerMap:
      map = ContainerMap()
  
      transformer_containers = [
          OPTTransformerContainer(self.model)
          for _ in range(self.model.num_layers)
      ]
  
      map.set_transformer_params(
          ["model.decoder.layers", "decoder.layers"],
          transformer_containers,
      )
  
      map.set_non_transformer_params(
          OPTNonTransformerContainer(self.model)
      )
  
      map.set_unmapped_params(["lm_head.weight"])
  
      return map
  
file_path: DeepSpeed/deepspeed/inference/v2/model_implementations/opt/policy.py
imports: from typing import Any
  from ...config_v2 import RaggedInferenceEngineConfig
  from ..inference_policy_base import ContainerMap, InferenceV2Policy
  from .container import OPTNonTransformerContainer, OPTTransformerContainer
  from .model import OPTInferenceModel
label_desc: Use image node_id 2 for calling the OPTPolicy obj's underlying member method code with example usage: obj.build_container_map() and returns: map