def get_hidden_state(
        self,
        outputs: BaseModelOutputWithPoolingAndCrossAttentions,
    ) -> torch.Tensor:
        """Override to return the last hidden state."""
        return outputs.last_hidden_state  # type: ignore
Override to return the last hidden state.
llmfoundry/models/llm_embed/finetune_embedding_model.py