code: def replace_arguments(config, world_size):
      return {
          # 1. reduce hidden size
          "self_attn.embed_dim": config.d_model // world_size,
          # 2. reduce number of heads
          "self_attn.num_heads": config.encoder_attention_heads
          // world_size,
      }
  
imports: from transformers.models.pegasus.modeling_pegasus import PegasusDecoderLayer, PegasusEncoderLayer
  from parallelformers.policies.base import Layer, Policy
  from parallelformers.transformers.modeling_bart import BartAttention_
  from parallelformers.utils.dist_utils import AllReduceLinear
label_desc: Use image node_id 1 for calling the PegasusEncoderPolicy obj's underlying member method code with example usage: obj.replace_arguments(config, world_size) and returns: dict