chunk_content: ntial) — The MLP head of the prompt encoder if inference_mode=False.
  lstm_head (torch.nn.LSTM) — The LSTM head of the prompt encoder if inference_mode=False and
  encoder_reparameterization_type="LSTM".
  token_dim (int) — The hidden embedding dimension of the base transformer model.
  input_size (int) — The input size of the prompt encoder.
  output_size (int) — The output size of the prompt encoder.
  hidden_size (int) — The hidden size of the prompt e