def ner_post_process(output_data, **kwargs):
    """Post-process data for NER task.

    Args:
        output_data (object): Model output to be post-processed.
        **kwargs: Additional arguments.

    Returns:
        object: Post-processed data.

    """
    import torch

    if isinstance(output_data, transformers.modeling_outputs.TokenClassifierOutput):
        logits = output_data.logits
    else:
        logits = output_data
    return torch.argmax(logits, dim=-1)
Post-process data for NER task.

    Args:
        output_data (object): Model output to be post-processed.
        **kwargs: Additional arguments.

    Returns:
        object: Post-processed data.

    
olive/data/component/post_process_data.py