o
    ॵiu                     @   s  d dl mZ d dlmZ erddlmZ ddlmZmZ ddl	m
Z
mZmZ ddlmZ ddlmZmZ dd	lmZmZmZmZmZmZ dd
lmZmZmZ ddlmZ ddlm Z  ddl!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z' ddl(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZL ddlMmNZNmOZO dS dgddgg dddgdgddgg d	g d
dgdgg dg ddZPd dlQZQeeReS d ePeTi deQjUeR< dS )    )TYPE_CHECKING)LazyImportModule   )Preprocessor)PREPROCESSORSbuild_preprocessor)ComposeToTensorFilter)WavToScp)LinearAECAndFbankAudioBrainPreprocessor)	LoadImage
load_image%ImageColorEnhanceFinetunePreprocessor%ImageInstanceSegmentationPreprocessorImageDenoisePreprocessorImageDeblurPreprocessor)#ImageClassificationMmcvPreprocessorImageRestorationPreprocessor'ControllableImageGenerationPreprocessor)
WavToLists)KanttsDataPreprocessor)$DiffusionImageGenerationPreprocessorOfaPreprocessorMPlugPreprocessorHiTeAPreprocessorMplugOwlPreprocessor+ImageCaptioningClipInterrogatorPreprocessor)$,DocumentSegmentationTransformersPreprocessor,FaqQuestionAnsweringTransformersPreprocessorFillMaskPoNetPreprocessor FillMaskTransformersPreprocessor#TextRankingTransformersPreprocessor*RelationExtractionTransformersPreprocessor)SentenceEmbeddingTransformersPreprocessor*TextClassificationTransformersPreprocessor'TextGenerationSentencePiecePreprocessor+TokenClassificationTransformersPreprocessorTextErrorCorrectionPreprocessorTextGenerationT5PreprocessorWordAlignmentPreprocessor&TextGenerationTransformersPreprocessorTokenize+WordSegmentationBlankSetToLabelPreprocessorMGLMSummarizationPreprocessor.ZeroShotClassificationTransformersPreprocessorTextGenerationJiebaPreprocessorSentencePiecePreprocessor"DialogIntentPredictionPreprocessorDialogModelingPreprocessorDialogStateTrackingPreprocessor#ConversationalTextToSqlPreprocessor"TableQuestionAnsweringPreprocessorNERPreprocessorVietNERPreprocessorThai WordSegmentationPreprocessorThai-TranslationEvaluationTransformersPreprocessorCanmtTranslationPreprocessor%DialogueClassificationUsePreprocessorSiameseUiePreprocessor*DocumentGroundedDialogGeneratePreprocessor+DocumentGroundedDialogRetrievalPreprocessor(DocumentGroundedDialogRerankPreprocessor-MachineReadingComprehensionForNERPreprocessor)ReadVideoData"MovieSceneSegmentationPreprocessorr   r   r   r   r   r   rC   rD   r   r   )'r   r    r!   r"   NLPTokenizerPreprocessorBaser#   r$   r%   r'   r&   r(   r)   r+   r,   r-   r*   r.   r/   CodeGeeXPreprocessorr0   r1   r2   r8   r9   r:   r3   r4   r5   r6   r7   r;   r<   r=   r>   r=   r?   r@   rA   rB   )basebuildercommonaudioasrvideoimagecvkwsttsmulti_modalnlpN__file__)module_specextra_objects)Vtypingr   modelscope.utils.import_utilsr   rG   r   rH   r   r   rI   r   r	   r
   rK   r   rJ   r   r   rM   r   r   r   r   r   r   rN   r   r   r   rO   r   rP   r   rQ   r   r   r   r   r   r   rR   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rL   rC   rD   _import_structuresys__name__globals__spec__modules r^   r^   U/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/preprocessors/__init__.py<module>   sH     C