o
    ॵi                     @   s  d dl mZ d dlmZ er#ddlmZ ddlmZ ddlm	Z	 ddl
mZ ddlmZ dd	lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddl m!Z! ddl"m#Z# ddl$m%Z% ddl&m'Z' ddl(m)Z) ddl*m+Z+ ddl,m-Z- ddl.m/Z/ ddl0m1Z1 ddl2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z: ddl;m<Z< ddl=m>Z> ddl?m@Z@ ddlAmBZB dd lCmDZDmEZE dd!lFmGZG dd"lHmIZI dd#lJmKZK dd$lLmMZM dd%lNmOZO dd&lPmQZQ dd'lRmSZS dd(lTmUZU dd)lVmWZW dd*lXmYZY dd+lZm[Z[ dd,l\m]Z] dd-l^m_Z_ dd.l`maZa d~S i d/d0gd1d2gd3d4gd5d6gd7d8gd9d:gd;d<gd=d>gd?d@gdAdBgdCdDgdEdFgdGdHgdIdJgdKdLgdMdNgdOdPgi dQdRgdSdTgdUdVgdWdXgdYdZgd[g d\d]d^gd_d`gdadbgdcddgdedfgdgdhdigdjdkgdldmgdndogdpdqgdrdsgdtgdugdvgdwgdxgdygdzgd{gd|gd}	Zbd d~lcZceedee d ebefi decjged< d~S )    )TYPE_CHECKING)LazyImportModule   )AutomaticPostEditingPipeline)ConversationalTextToSqlPipeline)TableQuestionAnsweringPipeline)DialogIntentPredictionPipeline)DialogModelingPipeline)DialogStateTrackingPipeline)DocumentSegmentationPipeline)ExtractiveSummarizationPipeline)PolyLMTextGenerationPipeline)&FasttextSequenceClassificationPipeline)FaqQuestionAnsweringPipeline)FeatureExtractionPipeline)FillMaskPipeline)InformationExtractionPipeline)InteractiveTranslationPipeline)NamedEntityRecognitionPipeline)TextRankingPipeline)SentenceEmbeddingPipeline)TextClassificationPipeline)SummarizationPipeline)$TranslationQualityEstimationPipeline)TextErrorCorrectionPipeline)WordAlignmentPipeline)TextGenerationPipelineTextGenerationT5PipelineSeqGPTPipelineChatGLM6bTextGenerationPipeline!ChatGLM6bV2TextGenerationPipelineQWenChatPipelineQWenTextGenerationPipelineLlama2TaskPipeline)FidDialoguePipeline)TokenClassificationPipeline)TranslationPipeline)CanmtTranslationPipeline)WordSegmentationPipelineWordSegmentationThaiPipeline)ZeroShotClassificationPipeline)MGLMTextSummarizationPipeline)CodeGeeXCodeTranslationPipeline)CodeGeeXCodeGenerationPipeline)GLM130bTextGenerationPipeline)TranslationEvaluationPipeline)"UserSatisfactionEstimationPipeline)SiameseUiePipeline)&DocumentGroundedDialogGeneratePipeline)'DocumentGroundedDialogRetrievalPipeline)$DocumentGroundedDialogRerankPipeline)LanguageIdentificationPipeline))MachineReadingComprehensionForNERPipeline)LLMPipelineautomatic_post_editing_pipeliner   #conversational_text_to_sql_pipeliner   polylm_text_generation_pipeliner   !dialog_intent_prediction_pipeliner   dialog_modeling_pipeliner	   dialog_state_tracking_pipeliner
   %fasttext_text_classification_pipeliner   document_segmentation_pipeliner   !extractive_summarization_pipeliner   faq_question_answering_pipeliner   feature_extraction_pipeliner   fill_mask_pipeliner   information_extraction_pipeliner    interactive_translation_pipeliner   !named_entity_recognition_pipeliner   text_ranking_pipeliner   sentence_embedding_pipeliner   summarization_pipeliner   !table_question_answering_pipeliner   text_classification_pipeliner   text_error_correction_pipeliner   word_alignment_pipeliner   text_generation_pipeline)r   r   r   r    r!   r"   r   r#   fid_dialogue_pipeliner$   token_classification_pipeliner%   translation_pipeliner&   canmt_translation_pipeliner'   'translation_quality_estimation_pipeliner   word_segmentation_pipeliner(   r)   !zero_shot_classification_pipeliner*    mglm_text_summarization_pipeliner+   "codegeex_code_translation_pipeliner,   !codegeex_code_generation_pipeliner-    glm130b_text_generation_pipeliner.   r/   r0   r1   r2   r4   r3   r5   r6   r7   )	translation_evaluation_pipeline%user_satisfaction_estimation_pipelinesiamese_uie_pipeline*document_grounded_dialog_generate_pipeline(document_grounded_dialog_rerank_pipeline+document_grounded_dialog_retrieval_pipelinelanguage_identification_pipline&machine_reading_comprehension_pipelinellm_pipelineN__file__)module_specextra_objects)htypingr   modelscope.utils.import_utilsr   r8   r   r9   r   rJ   r   r;   r   r<   r	   r=   r
   r?   r   r@   r   r:   r   r>   r   rA   r   rB   r   rC   r   rD   r   rE   r   rF   r   rG   r   rH   r   rK   r   rI   r   rS   r   rL   r   rM   r   rN   r   r   r   r   r    r!   r"   r#   rO   r$   rP   r%   rQ   r&   rR   r'   rT   r(   r)   rU   r*   rV   r+   rW   r,   rX   r-   rY   r.   rZ   r/   r[   r0   r\   r1   r]   r2   r_   r3   r^   r4   r`   r5   ra   r6   rb   r7   _import_structuresys__name__globals__spec__modules rn   rn   U/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/pipelines/nlp/__init__.py<module>   s(  (	%&'()+-/0245H