configuration.py 409 B

123456789101112131415161718192021
  1. # Copyright (c) Alibaba, Inc. and its affiliates.
  2. """UniTE model configuration"""
  3. from enum import Enum
  4. from modelscope.utils import logger as logging
  5. from modelscope.utils.config import Config
  6. logger = logging.get_logger()
  7. class InputFormat(Enum):
  8. SRC = 'src'
  9. REF = 'ref'
  10. SRC_REF = 'src-ref'
  11. class UniTEConfig(Config):
  12. def __init__(self, **kwargs):
  13. super().__init__(**kwargs)