def METHOD_NAME(
    self,
    source,
    id,
    padding_mask,
    features_only=True,
    remove_extra_tokens=True,
    classification_head_name=None,
):
    encoder_out = self.model(
        source,
        id=id,
        mode=Modality.TEXT,
        padding_mask=padding_mask,
        mask=False,
        features_only=features_only,
        remove_extra_tokens=remove_extra_tokens
    )
    logits = self.classification_heads[classification_head_name](encoder_out["x"])
    return logits, encoder_out