def handle_language_head(
        self,
        outputs: CausalLMOutputWithPast,
    ) -> torch.Tensor:
        """Handles `zero` tensor to avoid DDP unused parameters error."""
        return torch.sum(
            outputs.logits,  # type: ignore
        ) * 0  # This attaches the language head to the computation graph
Handles `zero` tensor to avoid DDP unused parameters error.
llmfoundry/models/llm_embed/modeling_llm_embed.py