chunk_content: nsformer_submodules=1,
  ...     num_attention_heads=12,
  ...     num_layers=12,
  ...     encoder_reparameterization_type="MLP",
  ...     encoder_hidden_size=768,
  ... )
  
  >>> prompt_encoder = PromptEncoder(config)
  Attributes:
  embedding (torch.nn.Embedding) — The embedding layer of the prompt encoder.
  mlp_head (torch.nn.Sequential) — The MLP head of the prompt encoder if inference_mode=False.
  lstm_head (torch.nn.LSTM) — The LSTM head of the prompt enc