def handle_language_head(
        self,
        outputs: BaseModelOutputWithPoolingAndCrossAttentions,
    ) -> torch.Tensor:
        """Override to skip language head handling."""
        return torch.tensor(
            0,
            dtype=torch.float32,
            device=outputs.last_hidden_state.device,  # type: ignore
        )
Override to skip language head handling.
llmfoundry/models/llm_embed/finetune_embedding_model.py