__init__.py 1.3 KB

123456789101112131415161718192021222324252627282930313233
  1. # Copyright (c) Alibaba, Inc. and its affiliates.
  2. from typing import TYPE_CHECKING
  3. from modelscope.utils.import_utils import LazyImportModule
  4. if TYPE_CHECKING:
  5. from .sequence_classification_trainer import SequenceClassificationTrainer
  6. from .csanmt_translation_trainer import CsanmtTranslationTrainer
  7. from .text_ranking_trainer import TextRankingTrainer
  8. from .text_generation_trainer import TextGenerationTrainer
  9. from .sentence_embedding_trainer import SentenceEmbeddingTrainer
  10. from .siamese_uie_trainer import SiameseUIETrainer
  11. from .translation_evaluation_trainer import TranslationEvaluationTrainer
  12. else:
  13. _import_structure = {
  14. 'sequence_classification_trainer': ['SequenceClassificationTrainer'],
  15. 'csanmt_translation_trainer': ['CsanmtTranslationTrainer'],
  16. 'text_ranking_trainer': ['TextRankingTrainer'],
  17. 'text_generation_trainer': ['TextGenerationTrainer'],
  18. 'sentence_emebedding_trainer': ['SentenceEmbeddingTrainer'],
  19. 'siamese_uie_trainer': ['SiameseUIETrainer'],
  20. 'translation_evaluation_trainer': ['TranslationEvaluationTrainer']
  21. }
  22. import sys
  23. sys.modules[__name__] = LazyImportModule(
  24. __name__,
  25. globals()['__file__'],
  26. _import_structure,
  27. module_spec=__spec__,
  28. extra_objects={},
  29. )