ModelScope/modelscope/models/nlp/unite/configuration_unite.py

# Copyright (c) Alibaba, Inc. and its affiliates.
"""UniTE model configuration"""

from enum import Enum

from modelscope.utils import logger as logging
from modelscope.utils.config import Config

logger = logging.get_logger(__name__)


class EvaluationMode(Enum):
    SRC = 'src'
    REF = 'ref'
    SRC_REF = 'src-ref'


class UniTEConfig(Config):

    def __init__(self, **kwargs):
        super().__init__(**kwargs)