code: def create_module(self, config=None):
      _config = config if config is not None else self.ds_model_config
      self.module = DeepSpeedGPTInference(
          _config, mp_group=self.mp_group
      )
      self.module.config.scale_attention = self.scale_attention
      return self.module
  
file_path: deepspeed/deepspeed/module_inject/containers/gpt2.py
imports: from .base import *
  from deepspeed.model_implementations.transformers.ds_gpt import DeepSpeedGPTInference
  from ..policy import TransformerPolicy
label_desc: Use image node_id 2 for calling the DS_GPT2Container obj's underlying member method code with example usage: obj.create_module(config) and returns: self