src/transformers/models/nezha/modeling_nezha.py/NezhaOnlyNSPHead/forward
class NezhaOnlyNSPHead:    def forward(self, pooled_output):
        seq_relationship_score = self.seq_relationship(pooled_output)
        return seq_relationship_score