o
    ߥiF                     @   s@  d dl mZ d dlmZ er_ddlmZ ddlmZmZm	Z	m
Z
mZmZmZmZmZ ddlmZmZ ddlmZ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m Z  ddl!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/ ddl0m1Z1m2Z2 ddl3m4Z4 ddl5m6Z6 ddl7m8Z8m9Z9 ddl:m;Z;m<Z<m=Z= ddl>m?Z? ddl@mAZA ddlBmCZCmDZDmEZE ddlFmGZGmHZHmIZI ddlJmKZKmLZLmMZM ddlNmOZO ddlPmQZQ ddlRmSZSmTZTmUZUmVZVmWZW ddlXmYZY dd lZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZambZbmcZc dd!ldmeZe dd"lfmgZg dd#lhmiZimjZjmkZkmlZlmmZm dd$lnmoZompZpmqZq dd%lrmsZsmtZt dd&lumvZvmwZwmxZxmyZymzZz dd'l{m|Z|m}Z}m~Z~mZmZ dd(lmZmZmZmZ dkS i d)d*gd+g dd,d-d.gd/d0gd1d2gd3d4gd5d6d7gd8d9gd:d;d<gd=g dd>g dd?d@gdAdBgdCdDgdEdFdGgdHdIdJgdKdLgi dMg ddNg d#dOdPdQgdRg ddSdTgdUdVgdWg ddXg ddYdZgd[d\gd]g dd^g d d_d`gdadbgdcddgdedfgdgg d$dhdigg d&g d'g d(djZd dklZeee dl eei dmeje< dkS )n    )TYPE_CHECKING)LazyImportModule   )BartForTextErrorCorrection)	BertForMaskedLMBertForTextRankingBertForSentenceEmbeddingBertForSequenceClassificationBertForTokenClassificationBertForDocumentSegmentation	BertModel
BertConfigSiameseUieModel)
BloomModelBloomForTextGeneration)CodeGeeXForCodeTranslationCodeGeeXForCodeGeneration)GLM130bForTextGeneration)CsanmtForTranslation)CanmtForTranslation)PolyLMForTextGeneration)DebertaV2ForMaskedLMDebertaV2Model)ChatGLMForConditionalGenerationChatGLMTokenizerChatGLMConfig) ChatGLM2ForConditionalGenerationChatGLM2TokenizerChatGLM2Config)GPTNeoModel)	GPT2Model)GPT3ForTextGenerationDistributedGPT3)GPTMoEForTextGenerationDistributedGPTMoE)TextClassificationHead)TransformersModel)	LSTMModel!LSTMForTokenClassificationWithCRF)MegatronBertConfigMegatronBertForMaskedLMMegatronBertModel)MGLMForTextSummarization)PalmForTextGeneration)PlugMentalConfigPlugMentalModel#PlugMentalForSequenceClassification)PoNetForMaskedLM
PoNetModelPoNetConfig)SpaceForDialogIntentSpaceForDialogModelingSpaceForDST)TableQuestionAnswering)StarForTextToSql)SbertForFaqQuestionAnsweringSbertForMaskedLMSbertForSequenceClassificationSbertForTokenClassification
SbertModel)T5ForConditionalGeneration)	ModelForFeatureExtractionModelForInformationExtractionModelForTextClassificationSingleBackboneTaskModelBaseModelForTextGenerationModelForTextRankingModelForTokenClassification"ModelForTokenClassificationWithCRF#ModelForMachineReadingComprehension)UniTEForTranslationEvaluation)UserSatisfactionEstimation)
VecoConfigVecoForMaskedLMVecoForSequenceClassificationVecoForTokenClassification	VecoModel)#DocumentGroundedDialogGenerateModel$DocumentGroundedDialogRetrievalModel!DocumentGroundedDialogRerankModel)XLMRobertaConfigXLMRobertaModel)LlamaForTextGenerationLlamaConfig
LlamaModelLlamaTokenizerLlamaTokenizerFast)Llama2ForTextGenerationLlama2ConfigLlama2ModelLlama2TokenizerLlama2TokenizerFast)QWenForTextGeneration
QWenConfig	QWenModelQWenTokenizerbartr   bertbloomr   r   csanmtr   canmtr   polylmr   codegeexr   r   glm_130br   
deberta_v2r   r   chatglmchatglm2headsr%   hf_transformersr&   gpt2r    gpt3r!   r"   gpt_moer#   r$   gpt_neor   
structbertvecolstmLSTMr(   megatron_bertmglmr,   palm_v2r-   plug_mentalponet
space_T_enr8   
space_T_cnr7   spacetask_modelssentence_embeddingSentenceEmbeddingT5r>   uniterH   userI   dgdsrR   rS   )xlm_robertallamallama2qwenN__file__)module_specextra_objects)typingr   modelscope.utils.import_utilsr   rb   r   rc   r   r   r   r	   r
   r   r   r   r   rd   r   r   rh   r   r   ri   r   re   r   rf   r   rg   r   rj   r   r   rk   r   r   r   rl   r   r   r   rr   r   ro   r    rp   r!   r"   rq   r#   r$   rm   r%   rn   r&   ru   r'   r(   rw   r)   r*   r+   rx   r,   ry   r-   rz   r.   r/   r0   r{   r1   r2   r3   r~   r4   r5   r6   r}   r7   r|   r8   rs   r9   r:   r;   r<   r=   r   r>   r   r?   r@   rA   rB   rC   rD   rE   rF   rG   r   rH   r   rI   rt   rJ   rK   rL   rM   rN   r   rO   rP   rQ   r   rR   rS   r   rT   rU   rV   rW   rX   r   rY   rZ   r[   r\   r]   r   r^   r_   r`   ra   _import_structuresys__name__globals__spec__modules r   r   R/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/models/nlp/__init__.py<module>   s   ,, !"#*15:;<ABCDFQRSTUZg