
    9i                        S SK JrJr   " S S\5      r " S S\5      r " S S\5      r " S S	\5      r0 \R                  \R                  S
4_\R                  \R                  S4_\R                  \R                  S4_\R                  \R                  S4_\R                  \R                  S4_\R                  \R                  S4_\R                  \R                  S4_\R                  \R                  S4_\R                   \R                  S4_\R"                  \R"                  S4_\R$                  \R&                  S4_\R(                  \R*                  S4_\R,                  \R,                  S4_\R.                  \R.                  S4_\R0                  \R0                  S4_\R2                  \R2                  S4_\R4                  \R4                  S4_0 \R6                  \R8                  S4_\R:                  \R<                  S4_\R>                  \R@                  S4_\RB                  \RB                  S4_\RD                  \RD                  S4_\RF                  \RF                  S4_\RH                  \R.                  S4_\RJ                  \RJ                  S4_\RL                  \RL                  S 4_\RN                  \RP                  S!4_\RR                  \RR                  S"4_\RT                  \RV                  S#4_\RX                  \RX                  S$4_\RZ                  \RZ                  S%4_\R\                  \R\                  S&4_\R^                  \R^                  S'4_\R`                  \R`                  S(4_E0 \Rb                  \Rb                  S)4_\Rd                  \Rf                  S*4_\Rh                  \Rh                  S+4_\Rj                  \Rj                  S,4_\Rl                  \Rl                  S-4_\Rn                  \Rn                  S.4_\Rp                  \Rp                  S/4_\Rr                  \Rr                  S04_\Rt                  \Rt                  S14_\Rv                  \Rv                  S24_\Rx                  \Rx                  S34_\Rz                  \Rz                  S44_\R|                  \R|                  S54_\R~                  \R~                  S64_\R                  \R                  S74_\R                  \R                  S84_\R                  \R                  S94_E0 \R                  \R                  S:4_\R                  \R                  S;4_\R                  \R                  S<4_\R                  \R                  S=4_\R                  \R                  S>4_\R                  \R                  S?4_\R                  \R                  S@4_\R                  \R                  SA4_\R                  \R                  SB4_\R                  \R                  SC4_\R                  \R                  SD4_\R                  \R                  SE4_\R                  \R                  SF4_\R                  \R                  SG4_\R                  \R                  SH4_\R                  \R                  SI4_\R                  \R                  SJ4_E0 \R                  \R                  SK4_\R                  \R                  SL4_\R                  \R                  SM4_\R                  \R                  SN4_\R                  \R                  SO4_\R                  \R                  SP4_\R                  \R                  SQ4_\R                  \R                  SR4_\R                  \R                  SS4_\R                  \R                  ST4_\R                  \R                  SU4_\R                  \R                  SV4_\R                  \R                  SW4_\R                  \R                  SX4_\R                  \R                  SY4_\R                  \R                  SZ4_\R                  \R                  S[4_E0 \R                  \R                  S\4_\R                  \R                  S]4_\R                  \R                  S^4_\R                  \R                  S_4_\R                  \R                  S`4_\R                  \R                  Sa4_\R                  \R                  Sb4_\R6                  \R                  Sc4_\R                  \R                  Sd4_\R                  \R                  Se4_\R                  \R                  Sf4_\R                  \R                  Sg4_\R                  \R                  Sh4_\R                  \R                  Si4_\R                  \R                  Sj4_\R                  \R                  Sk4_\R                  \R                  Sl4_E0 \R                  \R                  Sm4_\GR                   \GR                   Sn4_\GR                  \GR                  So4_\GR                  \GR                  Sp4_\GR                  \GR                  Sq4_\GR                  \GR                  Sr4_\GR
                  \GR
                  Ss4_\GR                  \GR                  St4_\GR                  \GR                  Su4_\GR                  \GR                  Sv4_\GR                  \GR                  Sw4_\GR                  \GR                  Sx4_\GR                  \GR                  Sy4_\GR                  \GR                  Sz4_\GR                  \GR                  S{4_\GR                  \GR                  S|4_\GR                  \GR                  S}4_E0 \GR                   \GR                   S~4_\GR"                  \GR"                  S4_\GR$                  \GR$                  S4_\GR&                  \GR&                  S4_\GR(                  \GR(                  S4_\GR*                  \GR*                  S4_\GR,                  \GR,                  S4_\GR.                  \GR.                  S4_\GR0                  \GR0                  S4_\GR2                  \GR2                  S4_\GR4                  \GR4                  S4_\GR6                  \GR8                  S4_\GR:                  \GR:                  S4_\GR<                  \GR<                  S4_\GR>                  \GR>                  S4_\GR@                  \GR@                  S4_\GRB                  \GRB                  S4_E\GRD                  \GRD                  S4\GRF                  \GRF                  S4\GRH                  \GRH                  S4\GRJ                  \GRJ                  S4\GRL                  \GRL                  S4\GRN                  \GRN                  S4\GRP                  \GRP                  S4\GRR                  \GRR                  S4\GRT                  \GRT                  S4\GRV                  \GRV                  S40
Er " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\\\\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      rg)    )FieldsTasksc                      \ rS rSrSrSrSrSrSrSr	Sr
S	rS
rSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSr Sr!S r"S!r#S"r$S#r%S$r&S%r'S&r(S'r)S(r*S)r+S*r,S+r-S,r.S-r/S.r0S/r1S0r2S1r3S2r4S3r5S4r6S5r7S6r8S7r9S8r:S9r;S:r<S;r=S<r>S=r?S>r@S?rAS@rBSArCSBrDSCrESDrFSErGSFrHSGrISHrJSIrKSJrLSKrMSLrNSMrOSNrPSOrQSPrRSQrSSRrTSSrUSTrVSUrWSVrXSWrYSXrZSYr[SZr\S[r]S\r^S]r_S^r`S_raS`rbSarcSbrdScreSdrfSergSfrhSgriShrjSirkSjrlSkrmSlrnSmroSnrpSorqSprrSqrsSrrtSsruStrvSurwSvrxSwrySxrzSyr{Szr|S{r}S|r~S}rS~rSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrg)Models   zNames for different models.

    Holds the standard model name to use for identifying different model.
This should be used to register models.

    Model name should only contain model information but not task information.
tinynas-detectiontinynas-damoyolo	detectionMaskScoringzimage-restorationzrealtime-object-detectionzrealtime-video-object-detectionscrfddepeClassificationModelEasyRobustModelbnextyolopv2nafnetcsrnetadaint
deeplpfnetrrdbcascade_mask_rcnn_swinmaskdino_swingpenzproduct-retrieval-embeddingzbody-2d-keypointszbody-3d-keypointshdformerHRNetCrowdCountingzface-2d-keypointszstar-68ldk-detectionzswinL-panoptic-segmentationzr50-panoptic-segmentationpassvitbFFTInpaintingzStablediffusion-Paintbyexamplezpgl-video-summarizationz!swinb-video-panoptic-segmentationz!swinb-video-instance-segmentationz+clip-it-language-guided-video-summarizationzswinL-semantic-segmentationz vitadapter-semantic-segmentationtext-driven-segmentationznewcrfs-depth-estimationzomnidata-normal-estimationzpanovit-layout-estimationzunifuse-depth-estimationzs2net-depth-estimationzdro-resnet18-depth-estimationz"raft-dense-optical-flow-estimationhuman-normal-estimationzresnet50-bertz)swinT-referring-video-object-segmentationferfairface
retinafacedamofdshop-segmentationmogfacemtcnnulfdrtsflirarcfacefacemaskflctinymogvideo-inpaintingzhuman-wholebody-keypointhand-staticface-human-hand-detectionface-emotionproduct-segmentationzimage-body-reshapingimage-skychangevideo-human-mattinghuman-reconstructiontext-texture-generationvideo-frame-interpolationvideo-object-segmentationvideo-deinterlacez!quadtree-attention-image-matchingz"loftr-image-local-feature-matchingzlightglue-image-matchingzvision-middlewarevidtvideo-stabilizationzreal-basicvsrzrcp-sceneflow-estimationzimage-casmvs-depth-estimationz image-geomvsnet-depth-estimationzvop-retrieval-modelzvop-retrieval-model-seddcolorzimage-probing-modeldefrcnimage-face-fusionzcontent-checkopen-vocabulary-detection-vildecbsrzmsrresnet-liteobject_detection_3dddpmOCRRecognitionOCRDetection	LoreModelimage-quality-assessment-mosimage-quality-assessment-man$image-quality-assessment-degradationm2fpnerf-recon-accnerf-recon-4knerf-recon-vq-compressionsurface-recon-commonzbts-depth-estimationvision-efficient-tuningbad-image-detectingcontrollable-image-generationlongshortnetfastinstz pedestrian-attribute-recognitionimage-try-onhuman-image-generationimage-view-transformimage-control-3d-portraitrifeanydoor self-supervised-depth-completionbertzpalm-v2
structbert
deberta_v2vecocsanmt-translationcanmtz	space-dstzspace-intentzspace-modelingz
space-T-enz
space-T-cntransformer-crfztoken-classification-for-nerz%transformer-crf-for-word-segmentationztransformer-softmaxlstm-crfzlstm-crf-for-word-segmentationzgcnn-crfbartgpt2gpt3zgpt-moezgpt-neoplugzbert-for-document-segmentationzponet-for-document-segmentationponetpolylmT5mglmcodegeexglm130bbloomunitezmegatron-bertuser-satisfaction-estimationzfid-plugzfid-T5lstmzxlm-robertatransformerszplug-mentaldoc2botpeerllamallama2	chatglm6bzchatglm2-6bzqwen-7bzsambert-hifiganspeech_frcrn_ans_cirm_16k)speech_zipenhancer_ans_multiloss_16k_basespeech_dfsmn_ansspeech_dfsmn_kws_char_farfield"speech_dfsmn_kws_char_farfield_iot"speech_kws_fsmn_char_ctc_nearfield(speech_mossformer_separation_temporal_8k)speech_mossformer2_separation_temporal_8k	kws-kwsbpzgeneric-asrz	wenet-asrzgeneric-itnzgeneric-puncz
generic-svztdnn-svzecapa-tdnn-svzcam++-svzeres2net-svzeres2netv2-svz	resnet-svz
res2net-svzeres2net-aug-svzscl-sdzscl-sd-xvectorz	cam++-lrezeres2net-lrezcluster-backendzrdino_ecapa-tdnn-svzsdpn_ecapa-svz
generic-lmaudio-quantizationzlaura-codecfunasrhifissr
unetvc_16kofaclip-multi-modal-embeddingzgemm-generative-multi-modalzrleg-generative-multi-modalmplugz!diffusion-text-to-image-synthesisz-multi-stage-diffusion-text-to-image-synthesislatent-text-to-video-synthesiszteam-multi-modal-similarityz video-clip-multi-modal-embeddingzprost-clip-text-video-retrievalmgeovldochiteasoonetefficient-diffusion-tuningcones2-inferencez	mplug-owlzclip-interrogatorstable-diffusionzstable-diffusion-xlvideocomposertext-to-360panorama-imagezimage-to-video-modelzvideo-to-video-modelunifoldzunifold-symmetry N)__name__
__module____qualname____firstlineno____doc__tinynas_detectiontinynas_damoyolor
   mask_scoringimage_restorationrealtime_object_detectionrealtime_video_object_detectionr   r   classification_modeleasyrobust_modelr   r   r   r   r   r   r   r   r   r   product_retrieval_embeddingbody_2d_keypointsbody_3d_keypointsbody_3d_keypoints_hdformercrowd_countingface_2d_keypointsstar_68ldk_detectionpanoptic_segmentationr50_panoptic_segmentationimage_reid_personimage_inpaintingimage_paintbyexamplevideo_summarizationvideo_panoptic_segmentationvideo_instance_segmentation#language_guided_video_summarizationswinL_semantic_segmentation vitadapter_semantic_segmentationtext_driven_segmentationnewcrfs_depth_estimationomnidata_normal_estimationpanovit_layout_estimationunifuse_depth_estimations2net_depth_estimationdro_resnet18_depth_estimation"raft_dense_optical_flow_estimationhuman_normal_estimationresnet50_bert#referring_video_object_segmentationr    r!   r"   r#   shop_segmentationr%   r&   r'   r(   r)   r*   r+   r,   r-   video_inpaintinghuman_wholebody_keypointhand_staticface_human_hand_detectionface_emotionproduct_segmentationimage_body_reshapingimage_skychangevideo_human_mattinghuman_reconstructiontext_texture_generationvideo_frame_interpolationvideo_object_segmentationvideo_deinterlace!quadtree_attention_image_matching"loftr_image_local_feature_matchinglightglue_image_matchingvision_middlewarer:   video_stabilizationreal_basicvsrrcp_sceneflow_estimationimage_casmvs_depth_estimation image_geomvsnet_depth_estimationvop_retrieval_modelvop_retrieval_model_ser<   image_probing_modelr=   image_face_fusioncontent_checkopen_vocabulary_detection_vildr@   msrresnet_literA   rB   ocr_recognitionocr_detectionlineless_table_recognitionimage_quality_assessment_mosimage_quality_assessment_man$image_quality_assessment_degradationrI   nerf_recon_accnerf_recon_4knerf_recon_vq_compressionsurface_recon_commonbts_depth_estimationvision_efficient_tuningbad_image_detectingcontrollable_image_generationrQ   rR    pedestrian_attribute_recognitionimage_try_onhuman_image_generationimage_view_transformimage_control_3d_portraitrW   rX    self_supervised_depth_completionrZ   palmr[   r\   r]   translationr_   	space_dstspace_intentspace_modeling
space_T_en
space_T_cntcrftoken_classification_for_ner	tcrf_wsegtransformer_softmaxlcrf	lcrf_wseggcnncrfrb   rc   rd   gpt_moegpt_neore   bert_for_dsponet_for_dsrf   rg   rh   ri   rj   rk   rl   rm   megatron_bertusefid_plugfid_T5ro   xlm_robertarp   plug_mentalrq   rr   rs   rt   
chatglm_6bchatglm2_6bqwen_7bsambert_hifiganrv   rw   rx   ry   rz   r{   r|   r}   	kws_kwsbpgeneric_asr	wenet_asrgeneric_itngeneric_punc
generic_svtdnn_svecapa_tdnn_svcampplus_sveres2net_sveres2netv2_sv	resnet_sv
res2net_sveres2net_aug_svscl_sdscl_sd_xvectorcampplus_lreeres2net_lrecluster_backendrdino_tdnn_svsdpn_sv
generic_lmaudio_quantizationlaura_codecr   r   r   r   clipgemmrlegr   	diffusionmulti_stage_diffusionvideo_synthesisteam
video_clipprostr   r   r   r   efficient_diffusion_tuningcones2_inference	mplug_owlclip_interrogatorstable_diffusionstable_diffusion_xlr   text_to_360panorama_imageimage_to_video_modelvideo_to_video_modelr   unifold_symmetry__static_attributes__r       S/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/modelscope/metainfo.pyr   r      s    ,)I L+ ;&G#ED0(EGFFFJD5#MD"?++!+)N+19 ;"&;3"E"E*W'"?'I$99!= ;95$C!)M&7#M*U'
CHJF+GED
CDGH
CG)9K ;!L11'O/17 ; ;+(K%)M&9+D/#M9$C!'I$/5G/F+#M%E"E%N/D&O"M!,#A #A +Q(D%N#M ;117/$C!!LH'I$!L51 ;DG'I$ DDJJD&KEI!L%NJJD#A 7I/D0IGDDDGGD2K4LEF	BDHGEE#M
(CHFDK!LKGDEFJKG (O ;0[-)%E")M&)M&/Y,0[-IKIK!LJG#MKK#MIJ'OF%NL!L'O)MGJ-KFGJ
C'D(D(DE3IK6O(D3J-EDEEF!=)I+)/#M ;11 G)r7  r   c                   4    \ rS rSrSrSrSrSrSrSr	Sr
S	rS
rg)
TaskModelsi  text-classificationtoken-classificationinformation-extraction	fill-maskfeature-extractiontext-generationtext-rankingzmachine-reading-comprehensionr   N)r   r   r   r   text_classificationtoken_classificationinformation_extraction	fill_maskfeature_extractiontext_generationtext_rankingmachine_reading_comprehensionr6  r   r7  r8  r:  r:    s/    /15I-'O!L$C!r7  r:  c                   @    \ rS rSrSrSrSrSrSrSr	Sr
S	rS
rSrSrSrg)Headsi  r;  r>  zbert-mlmzroberta-mlmzxlm-roberta-mlmr<  r=  r@  rA  ra   r`   r   N)r   r   r   r   rB  rE  bert_mlmroberta_mlmxlm_roberta_mlmrC  rD  rG  rH  lstm_crftransformer_crfr6  r   r7  r8  rK  rK    s>     0IHK'O15'O!LH'Or7  rK  c                      \ rS rSrSrSrSrSrSrSr	Sr
S	rS
rSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSr Sr!S r"S!r#S"r$S#r%S#r&S$r'S%r(S&r)S'r*S(r+S)r,S*r-S+r.S,r/S-r0S.r1S/r2S0r3S1r4S2r5S3r6S4r7S5r8S6r9S7r:S8r;S9r<S:r=S;r>S<r?S=r@S>rAS?rBS@rCSArDSBrESCrFSDrGSErHSFrGSGrISHrJSIrKSJrLSKrMSLrNSMrOSNrPS/r2SOrQSPrRSQrSSRrTSSrUSTrVSUrWSVrXSWrYSXrZSYr[SZr\S[r]S\r^S]r_S^r`S_raS`rbSarcSbrdScreSdrfSergSfrhSgriShrjSirkSjrlSkrmSlrnSmroSnrpSorqSprrSqrsSrrtSsruStrvSurwSvrxSwrySxrzSyr{Szr|S{r}S|r~S}rS~rSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSGr SGrGS GrGSGrGSGrGSGrGSGrGSGrGSGrGSGr	GSGr
GS	GrGS
GrGSGrGSGrGSGrGSGrGSGrGSGrGSGrGSGrGSGrGSGrGSGrGSGrGSGrGSGrGSGrGSGrGSGrGSGrGSGrGSGr GSGr!GS Gr"GS!Gr#GS"Gr$GS#Gr%GS$Gr&GS%Gr'GS&Gr(GS'Gr)GS(Gr*GS)Gr+GS*Gr,GS+Gr-GS,Gr.GS-Gr/GS.Gr0GS/Gr1GS0Gr2GS1Gr3GS2Gr4GS3Gr5GS4Gr6GS5Gr7GS6Gr8Gg7(8  	Pipelinesi*  av  Names for different pipelines.

    Holds the standard pipeline name to use for identifying different pipeline.
This should be used to register pipelines.

    For pipeline which support different models and implements the common function, we
should use task name for this pipeline.
    For pipeline which support only one model, we should use ${Model}-${Task} as its name.
zpipeline-templatezunet-image-mattingzunet-universal-mattingznafnet-image-denoiseznafnet-image-deblurzmasactrl-image-editingz!freeu-stable-diffusion-text2imagezunet-person-image-cartoonzresnet18-ocr-detectionzdla34-table-recognitionzlore-lineless-table-recognitionz resnet18-license-plate-detectionz"resnet18-card-detection-correctionzTAdaConv_action-recognitionzresnet101-animal-recognitionzresnet101-general-recognitionzcmdssl-r2p1d_video_embeddingzhicossl-s3dg-video_embeddingz"hrnetv2w32_body-2d-keypoints_imagez!canonical_body-3d-keypoints_videoz"hrnetv2w18_hand-2d-keypoints_imagezresnet18-human-detectionztbs-detectionzvit-object-detectionzabnormal-object-detectionz%mobilenet_face-2d-keypoints_alignmentzu2net-salient-detectionzres2net-salient-detectionzres2net-camouflaged-detectionzuhdm-image-demoireingimage-classificationz!resnet-face-detection-scrfd10gkpszmanual-face-liveness-flirzmanual-face-liveness-flxcz!resnet-card-detection-scrfd34gkpszmanual-face-detection-ulfdzmanual-face-detection-tinymogz'vgg19-facial-expression-recognition-ferz&manual-facial-landmark-confidence-flcmzfacial-68ldk-detectionz,resnet34-face-attribute-recognition-fairfacez"resnet50-face-detection-retinafacez+resnet101-face-detection-cvpr22papermogfacezmanual-face-detection-mtcnnzlive-categoryz-vit-base_image-classification_ImageNet-labelsz.vit-base_image-classification_Dailylife-labelsz3nextvit-small_image-classification_Dailylife-labelsz*convnext-base_image-classification_garbagez0bnext-small_image-classification_ImageNet-labelsz'yolopv2_image-driving-percetion_bdd100kzcommon-image-classificationzcsrnet-image-color-enhancezadaint-image-color-enhancezdeeplpf-image-color-enhancezvirtual-try-onzunet-image-colorizationzAAMS-style-transferzrrdb-image-super-resolutionzimage-super-resolution-pasdzrrdb-image-debandingzgan-face-image-generationz$resnet50-product-retrieval-embeddingz1cspnet_realtime-video-object-detection_streamyolozir101-face-recognition-cfglintzir-face-recognition-ood-rtsz"manual-face-quality-assessment-fqazir-face-recognition-rtszmanual-face-recognition-frirzmanual-face-recognition-frfmzir50-face-recognition-arcfacez resnet-face-recognition-facemaskz resnet50-image-classification-ccz2cascade-mask-rcnn-swin-image-instance-segmentationz)maskdino-swin-image-instance-segmentationzimage-to-image-translationzvideo-categoryzconvnextTiny-ocr-recognitionzgpen-image-portrait-enhancementzimage-to-image-generationz!yolox_image-object-detection-autozyolox-pai_hand-detectionzunet-skin-retouchingzresnet50-face-reconstructionzHRN-head-reconstructionzHRN-text-to-headztinynas-classificationzeasyrobust-classificationr   zhrnet-crowd-countingzResNetC3D-action-detectionz)ostrack-vitb-video-single-object-trackingz,procontext-vitb-video-single-object-trackingzvideo-multi-object-trackingzimage-panoptic-segmentation!googlenet_pgl_video_summarizationzclip-it-video-summarizationzimage-semantic-segmentationzimage-depth-estimationzimage-normal-estimationzindoor-layout-estimationzimage-local-feature-matchingzvideo-depth-estimationzpanorama-depth-estimationzpanorama-depth-estimation-s2netzdense-optical-flow-estimationzpassvitb-image-reid-personzfft-inpaintingzstablediffusion-paintbyexamplezimage-inpainting-sdv2r   z&resnet50-bert-movie-scene-segmentationr$   r.   z'hrnetw48_human-wholebody-keypoint_imagezpatchshift-action-recognitionr/   r0   r1   r2   zflow-based-body-reshaping#referring-video-object-segmentationr3   r4   r5   r6   zvision-middleware-multi-taskr:   r7   r8   r9   zimage-matchingzimage-matching-fastr;   z#realbasicvsr-video-super-resolutionzpointcloud-sceneflow-estimationz!image-multi-view-depth-estimationzvideo-panoptic-segmentationzvideo-instance-segmentationzvop-video-text-retrievalzvop-video-text-retrieval-sezddcolor-image-colorizationzimage-structured-model-probingimage-fewshot-detectionr>   r?   z ddpm-image-semantic-segmentationzvideo-colorizationzmdm-motion-generationzmobile-image-super-resolutionzm2fp-image-human-parsingzobject-detection-3d-deperJ   rK   rL   rM   rO   rP   zfast-instance-segmentationrF   rG   rH   rN   zimage-bts-depth-estimationzimage-depth-estimation-marigoldz/resnet50_pedestrian-attribute-recognition_imager   rS   rT   zhuman3d-renderzhuman3d-animationrU   rV   zrife-video-frame-interpolationrX   zimage-to-3drY   r   zautomatic-post-editingztranslation-quality-estimationzdomain-classificationzsentence-similarityzword-segmentationzmultilingual-word-segmentationzword-segmentation-thaizpart-of-speechznamed-entity-recognitionznamed-entity-recognition-thaiznamed-entity-recognition-vietr@  zfid-dialogueztext2text-generationzsentiment-analysiszsentiment-classificationr;  r>  fill-mask-ponetr^   canmt-translationzinteractive-translationnlizdialog-intent-predictionzdialog-modelingzdialog-state-trackingzzero-shot-classificationtext-error-correctionword-alignmentzplug-generationzgpt3-generationzpolylm-text-generationzgpt-moe-generationzfaq-question-answeringconversational-text-to-sqlz!table-question-answering-pipelinesentence-embeddingrA  mgeo-rankingzrelation-extractiondocument-segmentationzextractive-summarizationr?  zmglm-text-summarizationzcodegeex-code-translationzcodegeex-code-generationzglm130b-text-generationtranslation_en_to_detranslation_en_to_rotranslation_en_to_frr<  ztranslation-evaluationrn   zsiamese-uie"document-grounded-dialog-retrievaldocument-grounded-dialog-rerank!document-grounded-dialog-generatelanguage_identification%machine-reading-comprehension-for-nerllmzsambert-hifigan-ttszspeech-dfsmn-aec-psm-16krv   rw   speech_dfsmn_ans_psm_48k_causalry   speech-separationr~   zasr-wenet-inferencezitn-inferencezspeaker-diarization-inferencezvad-inferencezfunasr-speech-separationzspeaker-verificationzspeaker-verification-tdnnzspeaker-verification-rdinozspeaker-verification-sdpnzspeaker-verification-eres2netzspeaker-verification-eres2netv2zspeaker-verification-resnetzspeaker-verification-res2netzspeech-language-recognitionz$speech-language-recognition-eres2netzspeaker-change-locatingz&speaker-diarization-dialogue-detectionz3speaker-diarization-semantic-speaker-turn-detectionzsegmentation-clusteringzlanguage-score-predictionzspeech-timestamp-inferencer   zaudio-quantization-inferencezlaura-codec-tts-inferencez!speech-super-resolution-inferencezvoice-conversionzimage-captioningzmulti-modal-embeddingz generative-multi-modal-embeddingzvisual-question-answeringzvisual-groundingzvisual-entailmentzmulti-modal-similarityztext-to-image-synthesiszvideo-multi-modal-embeddingzprost-text-video-retrievalr   zimage-text-retrievalzofa-ocr-recognitionzofa-asrz
ofa-sudokuzofa-text2sqlzvideo-captioningzvideo-question-answeringzdiffusers-stable-diffusiondisco_guided_diffusionzdocument-vl-embeddingzchinese-stable-diffusionr   r   z"gridvlp-multi-modal-classificationzgridvlp-multi-modal-embeddingzsoonet-video-temporal-groundingr   zmultimodal-dialoguezllama2-text-generation-pipelinez$llama2-text-generation-chat-pipelinezimage-to-video-task-pipelinezvideo-to-video-pipelinezunifold-protein-structurezfunasr-pipeliner   N(9  r   r   r   r   r   pipeline_templateportrait_mattinguniversal_mattingimage_denoiseimage_deblurimage_editing!freeu_stable_diffusion_text2imageperson_image_cartoonr   table_recognitionr   license_plate_detectioncard_detection_correctionaction_recognitionanimal_recognitiongeneral_recognitioncmdssl_video_embeddinghicossl_video_embeddingr   r   hand_2d_keypointshuman_detectiontbs_detectionobject_detectionabnormal_object_detectionr   salient_detectionsalient_boudary_detectioncamouflaged_detectionimage_demoireimage_classificationface_detectionface_liveness_irface_liveness_rgbface_liveness_xccard_detectionulfd_face_detectiontinymog_face_detectionfacial_expression_recognitionfacial_landmark_confidencefacial_68ldk_detectionface_attribute_recognitionretina_face_detectionmog_face_detectionmtcnn_face_detectionlive_categorygeneral_image_classificationdaily_image_classification(nextvit_small_daily_image_classification*convnext_base_image_classification_garbage bnext_small_image_classification'yolopv2_image_driving_percetion_bdd100kcommon_image_classificationimage_color_enhanceadaint_image_color_enhancedeeplpf_image_color_enhancevirtual_try_onimage_colorizationimage_style_transferimage_super_resolutionimage_super_resolution_pasdimage_debandingface_image_generationr   r   face_recognitionface_recognition_oodface_quality_assessmentface_recognition_onnx_irface_recognition_onnx_fmarc_face_recognitionmask_face_recognitionr   image_instance_segmentationmaskdino_instance_segmentationimage2image_translationvideo_categoryr   image_portrait_enhancementimage_to_image_generationimage_object_detection_autohand_detectionskin_retouchingface_reconstructionhead_reconstructiontext_to_headtinynas_classificationeasyrobust_classificationr   r   action_detectionvideo_single_object_tracking'video_single_object_tracking_procontextvideo_multi_object_trackingimage_panoptic_segmentationr   r   image_semantic_segmentationimage_depth_estimationimage_normal_estimationindoor_layout_estimationimage_local_feature_matchingvideo_depth_estimationpanorama_depth_estimationpanorama_depth_estimation_s2netdense_optical_flow_estimationr   r   r   image_inpainting_sdv2r   movie_scene_segmentationr   r   r   pst_action_recognitionr   r   r   r   r   r   r   r   r   r   vision_middleware_multi_taskr:   r   r   r   image_matchingimage_matching_fastr   video_super_resolutionpointcloud_sceneflow_estimation!image_multi_view_depth_estimationr   r   vop_retrievalvop_retrieval_seddcolor_image_colorizationimage_structured_model_probingimage_fewshot_detectionr   r    ddpm_image_semantic_segmentationvideo_colorizationmotion_generattionmobile_image_super_resolutionimage_human_parsingobject_detection_3d_deper   r   r   r   r   r   fast_instance_segmentationr   r   r   r   image_bts_depth_estimationimage_depth_estimation_marigoldr   r2  r   r   human3d_renderhuman3d_animationr   r   rife_video_frame_interpolationrX   image_to_3dr   r   automatic_post_editingtranslation_quality_estimationdomain_classificationsentence_similarityword_segmentationmultilingual_word_segmentationword_segmentation_thaipart_of_speechnamed_entity_recognitionnamed_entity_recognition_thainamed_entity_recognition_vietrG  fid_dialoguetext2text_generationsentiment_analysissentiment_classificationrB  rE  fill_mask_ponetcsanmt_translationcanmt_translationinteractive_translationrY  dialog_intent_predictiondialog_modelingdialog_state_trackingzero_shot_classificationtext_error_correctionword_alignmentplug_generationgpt3_generationpolylm_text_generationgpt_moe_generationfaq_question_answeringconversational_text_to_sql!table_question_answering_pipelinesentence_embeddingrH  mgeo_rankingrelation_extractiondocument_segmentationextractive_summarizationrF  mglm_text_summarizationcodegeex_code_translationcodegeex_code_generationglm130b_text_generationr`  ra  rb  rC  translation_evaluationuser_satisfaction_estimationsiamese_uie"document_grounded_dialog_retrievaldocument_grounded_dialog_rerank!document_grounded_dialog_generaterf  %machine_reading_comprehension_for_nerrh  sambert_hifigan_ttsspeech_dfsmn_aec_psm_16krv   rw   ri  ry   speech_separationr  asr_wenet_inferenceitn_inferencespeaker_diarization_inferencevad_inferencefunasr_speech_separationspeaker_verificationspeaker_verification_tdnnspeaker_verification_rdinospeaker_verification_sdpnspeaker_verification_eres2netspeaker_verification_eres2netv2speaker_verification_resnetspeaker_verification_res2netspeech_language_recognition$speech_language_recognition_eres2netspeaker_change_locating&speaker_diarization_dialogue_detection3speaker_diarization_semantic_speaker_turn_detectionsegmentation_clusteringlm_inferencespeech_timestamp_inferencer!  audio_quantization_inferencelaura_codec_tts_inference!speech_super_resolution_inferencevoice_conversionimage_captioningmulti_modal_embedding generative_multi_modal_embeddingvisual_question_answeringvisual_groundingvisual_entailmentmulti_modal_similaritytext_to_image_synthesisvideo_multi_modal_embeddingprost_text_video_retrievalr   image_text_retrievalofa_ocr_recognitionofa_asr
ofa_sudokuofa_text2sqlvideo_captioningvideo_question_answeringdiffusers_stable_diffusionrk  document_vl_embeddingchinese_stable_diffusionr-  text_to_video_synthesis"gridvlp_multi_modal_classificationgridvlp_multi_modal_embeddingsoonet_video_temporal_groundingr,  multimodal_dialoguellama2_text_generation_pipeline$llama2_text_generation_chat_pipelineimage_to_video_task_pipelinevideo_to_video_pipelineprotein_structurefunasr_pipeliner6  r   r7  r8  rR  rR  *  s    ,+0*M(L,M(K%6,M1!B@ D679;<<;<0O#M- ;?1 ;;+M18N2328N6<$M!!I5!O@F8#M#R !Q/d,1].'Y$.W+"?6!="?%N20:"?,O7"H&Y#78B4==:>6M"V%P":#M%N4O!B ;"E/N,O83%L5 ;++N3#N .\+"?"?=*G'"?579#A 5 ;&G#$C!4';39G+)H<K ;!L16*O''O/17#A D ; ;+%N//B&G#(K%"?"?.M4!=%E"7+%E"'I$-0$C!49%N#M ;1/$C!!=#A #A +Q(7!=&G#'X$ ;!L5%N+1 ;%E"GK'I$7 6%E"3/+%E"5%N9$C!$C!'O!L1-9/I'O-+7
C9'O393%N'O'O5-5!=(K%-!L!L/39-7 ;9711115#A K)M&&G#(K%7,S)
C 09 ;0[-&G#%E"+I/#M$C!#M91 ;!= ;$C!&G#"?#A "?+Q(7-U*:o77.L!=-#A  ;(K%) *3'I$ ;)+57"?!=#M1/GJ!L)9!=539)>)M&$C!&G#!=/&G#+Q(#A 7 4 (Or7  rR  z.damo/nlp_corom_sentence-embedding_english-basez&damo/mgeo_address_ranking_chinese_basez+damo/nlp_corom_passage-ranking_english-basez2damo/nlp_structbert_word-segmentation_chinese-basez/damo/nlp_structbert_part-of-speech_chinese-basez9damo/nlp_raner_named-entity-recognition_chinese-base-newsz.damo/nlp_bert_relation-extraction_chinese-basez4damo/nlp_structbert_sentence-similarity_chinese-basez&damo/nlp_canmt_translation_zh2en_largez!damo/nlp_csanmt_translation_zh2enz$damo/nlp_structbert_nli_chinese-basez9damo/nlp_structbert_sentiment-classification_chinese-basezdamo/cv_unet_image-mattingzdamo/cv_unet_universal-mattingz damo/cv_resnet18_human-detectionz!damo/cv_vit_object-detection_cocoz!damo/cv_nafnet_image-denoise_siddz!damo/cv_nafnet_image-deblur_goprozdamo/cv_masactrl_image-editingz)damo/cv_dut-raft_video-stabilization_basez3damo/cv_realbasicvsr_video-super-resolution_videolqz-damo/nlp_palm2.0_text-generation_chinese-basez9damo/nlp_structbert_zero-shot-classification_chinese-basezdamo/nlp_space_dialog-modelingz$damo/nlp_space_dialog-state-trackingz$damo/nlp-convai-text2sql-pretrain-cnz*DAMO_ConvAI/nlp_convai_generation_pretrainzdamo/nlp_convai_rerank_pretrainz)DAMO_ConvAI/nlp_convai_retrieval_pretrainz+damo/nlp_bart_text-error-correction_chinesez$damo/ofa_image-caption_coco_large_enz/damo/multi-modal_hitea_video-captioning_base_enz1damo/cv_unet_person-image-cartoon_compound-modelsz.damo/cv_resnet18_ocr-detection-line-level_damoz9damo/cv_dla34_table-structure-recognition_cycle-centernetz;damo/cv_resnet-transformer_table-structure-recognition_lorez6damo/multi-modal_convnext-roberta-base_vldoc-embeddingz-damo/cv_resnet18_license-plate-detection_damoz damo/cv_resnet18_card_correctionzdamo/nlp_veco_fill-mask-largez&damo/pert_feature-extraction_base-testz#damo/cv_TAdaConv_action-recognitionz.damo/cv_ResNetC3D_action-detection_detection2dzdamo/cv_resnet50_live-categoryzdamo/cv_resnet50_video-categoryz)damo/multi-modal_clip-vit-base-patch16_zhzHdamo/multi-modal_gemm-vit-large-patch14_generative-multi-modal-embeddingz>damo/multi-modal_team-vit-large-patch14_multi-modal-similarityz2damo/mplug_visual-question-answering_coco_large_enz7damo/multi-modal_hitea_video-question-answering_base_enzdamo/cv_r2p1d_video_embeddingz.damo/cv_diffusion_text-to-image-synthesis_tinyzdamo/text-to-video-synthesisz*damo/cv_hrnetv2w32_body-2d-keypoints_imagez)damo/cv_canonical_body-3d-keypoints_videoz3damo/cv_hrnetw18_hand-pose-keypoints_coco-wholebodyz(damo/cv_resnet_carddetection_scrfd34gkpsz!damo/cv_resnet50_content-check_ccz3damo/cv_resnet101_face-detection_cvpr22papermogfacez!damo/cv_manual_face-liveness_flirz%damo/cv_ir101_facerecognition_cfglintz/damo/cv_vgg19_facial-expression-recognition_ferz4damo/cv_resnet34_face-attribute-recognition_fairfacez-damo/cv_mobilenet_face-2d-keypoints_alignmentz*damo/cv_manual_face-quality-assessment_fqaz*damo/multi_modal_clip_vtretrival_msrvtt_53z'damo/multi_modal_clip_vtretrieval_prostz)damo/cv_csrnet_image-color-enhance-modelsz"damo/cv_daflow_virtual-try-on_basez"damo/cv_ddcolor_image-colorizationzdamo/cv_unet_video-colorizationz/damo/cv_swin-b_image-instance-segmentation_cocoz0damo/cv_yolopv2_image-driving-perception_bdd100kz-damo/cv_newcrfs_image-depth-estimation_indoorz6Damo_XR_Lab/cv_omnidata_image-normal-estimation_normalz0Damo_XR_Lab/cv_human_monocular-normal-estimationz(damo/cv_panovit_indoor-layout-estimationz2damo/cv_dro-resnet18_video-depth-estimation_indoorz)damo/cv_unifuse_panorama-depth-estimationz-Damo_XR_Lab/cv_raft_dense-optical-flow_thingszEDamo_XR_Lab/cv_resnet-transformer_local-feature-matching_outdoor-dataz damo/cv_aams_style-transfer_damoz!damo/cv_gan_face-image-generationz#damo/cv_rrdb_image-super-resolutionzdamo/cv_rrdb_image-debandingz'damo/cv_gpen_image-portrait-enhancementz-damo/cv_resnet50_product-bag-embedding-modelsz-damo/cv_latent_diffusion_image2image_generatez6damo/cv_vit-base_image-classification_Dailylife-labelsz)damo/cv_yolox_image-object-detection-autoz1damo/cv_convnextTiny_ocr-recognition-general_damozdamo/cv_unet_skin-retouchingz7damo/nlp_structbert_faq-question-answering_chinese-basez#damo/cv_hrnet_crowd-counting_dcanetz1damo/cv_vitb_video-single-object-tracking_ostrackz)damo/cv_passvitb_image-reid-person_marketz+damo/cv_vitl16_segmentation_text-driven-segz7damo/cv_resnet50-bert_video-scene-segmentation_movienetz$damo/cv_vitb16_segmentation_shop-segzdamo/cv_fft_inpainting_lamaz)damo/cv_stable-diffusion_paint-by-examplezDdienstag/cv_controlnet_controllable-image-generation_nine-annotatorszdamo/cv_video-inpaintingz$damo/cv_effnetv2_video-human-mattingz(damo/cv_hrnet_image-human-reconstructionz(damo/cv_diffuser_text-texture-generationz&damo/cv_raft_video-frame-interpolationzdamo/cv_unet_video-deinterlacez/damo/cv_hrnetw48_human-wholebody-keypoint_imagezdamo/cv_mobileface_hand-staticz)damo/cv_nanodet_face-human-hand-detectionzdamo/cv_face-emotionz"damo/cv_F3Net_product-segmentationz2damo/cv_swin-t_referring_video-object-segmentationz)damo/cv_googlenet_pgl-video-summarizationzdamo/cv_hrnetocr_skychangez<damo/nlp_unite_mup_translation_evaluation_multilingual_largez(damo/cv_rdevos_video-object-segmentationz2damo/cv_yolov5_video-multi-object-tracking_fairmotz2damo/cv_casmvs_multi-view-depth-estimation_generalz*damo/cv_resnet101_detection_fewshot-defrcnz&damo/cv_flow-based-body-reshaping_damoz#damo/cv_unet-image-face-fusion_damoz1damo/cv_quadtree_attention_image-matching_outdoorz6damo/cv_resnet_image-quality-assessment-mos_youtubeUGCz5damo/cv_resnet50_image-quality-assessment_degradationz=damo/cv_vitb16_classification_vision-efficient-tuning-adapterz damo/cv_object-detection-3d_depez(damo/cv_mobilenet-v2_bad-image-detectingz.damo/cv_nerf-3d-reconstruction-accelerate_damoz+damo/cv_nerf-3d-reconstruction-4k-nerf_damoz2damo/cv_nerf-3d-reconstruction-vq-compression_damoz%damo/cv_surface-reconstruction-commonz,damo/nlp_structbert_siamese-uie_chinese-basez7damo/cv_resnet50_pedestrian-attribute-recognition_imagez6damo/cv_diffusion_text-to-360panorama-image_generationzdamo/cv_SAL-VTON_virtual-try-onz&damo/cv_FreqHPT_human-image-generationz"damo/cv_3d-human-synthesis-libraryzdamo/cv_3d-human-animationzdamo/cv_image-view-transformz/damo/cv_vit_image-control-3d-portrait-synthesisz%damo/self-supervised-depth-completionc                   \    \ rS rSrSrSrSrSrSrSr	Sr
S	rS
rSrSrSrSrSrSrSrSrSrSrg)
CVTrainersi  zimage-instance-segmentationzimage-portrait-enhancementzvideo-summarizationzmovie-scene-segmentationzface-detection-scrfdzcard-detection-scrfdzimage-inpaintingrU  zimage-classification-teamrS  rV  ocr-recognitionzocr-detection-dbrJ   rK   zaction-detectionrN   rY   r   N)r   r   r   r   r  r  r   r  face_detection_scrfdcard_detection_scrfdr   r   image_classification_teamr  r  r   ocr_detection_dbr   r   r  r   r   r6  r   r7  r8  rP  rP    sk    "?!=/911)*O' ;17'O)%N#M)7'I$r7  rP  c                   \    \ rS rSrSrSrSrSrSrSr	Sr
S	rS
rSrSrSrSrSrSrSrSrSrSrg)NLPTrainersi  zbert-sentiment-analysiszdialog-modeling-trainerzdialog-intent-trainerznlp-base-trainerznlp-veco-trainerznlp-text-ranking-trainerznlp-sentence-embedding-trainerztext-generation-trainerznlp-plug-trainerznlp-gpt3-trainerzfaq-question-answering-trainerznlp-gpt-moe-trainerz table-question-answering-trainerz)document-grounded-dialog-generate-trainerz'document-grounded-dialog-rerank-trainerz*document-grounded-dialog-retrieval-trainerzsiamese-uie-trainerztranslation-evaluation-trainerr   N)r   r   r   r   bert_sentiment_analysisdialog_modeling_trainerdialog_intent_trainernlp_base_trainernlp_veco_trainernlp_text_ranking_trainernlp_sentence_embedding_trainertext_generation_trainernlp_plug_trainergpt3_trainerfaq_question_answering_trainergpt_moe_trainer table_question_answering_trainer)document_grounded_dialog_generate_trainer'document_grounded_dialog_rerank_trainer*document_grounded_dialog_retrieval_trainersiamese_uie_trainertranslation_evaluation_trainerr6  r   r7  r8  rW  rW    sl    773))9%E"7)%L%E"+O'I$0[-.W+1]./%E"r7  rW  c                   @    \ rS rSrSrSrSrSrSrSr	Sr
S	rS
rSrSrSrg)MultiModalTrainersi  r   r   r   zmgeo-ranking-trainerr   r   zlora-diffusionzlora-diffusion-xlzdreambooth-diffusionzcustom-diffusionr   r   N)r   r   r   r   clip_multi_modal_embeddingr   r   mgeo_ranking_trainerr,  r0  lora_diffusionlora_diffusion_xldreambooth_diffusioncustom_diffusionr-  r6  r   r7  r8  rk  rk    sA    !=
CE1!=)%N+1))r7  rk  c                   ,    \ rS rSrSrSrSrSrSrSr	Sr
g	)
AudioTrainersi"  rv   ry   r{   zspeech-kantts-trainerzspeech-asr-trainerrj  r   N)r   r   r   r   rv   ry   r{   speech_kantts_trainerspeech_asr_trainerr  r6  r   r7  r8  rs  rs  "  s&     ;%E")M&3-+r7  rs  c                   0    \ rS rSrSrSrSr\S 5       rSr	g)Trainersi+  a  Names for different trainer.

    Holds the standard trainer name to use for identifying different trainer.
This should be used to register trainers.

    For a general Trainer, you can use EpochBasedTrainer.
    For a model specific Trainer, you can use ${ModelName}-${Task}-trainer.
trainerr	   c                 l   U [        [        5      ;   d!  U [        [        5      R                  5       ;   a  [        R                  $ U [        [
        5      ;   d!  U [        [
        5      R                  5       ;   a  [        R                  $ U [        [        5      ;   d!  U [        [        5      R                  5       ;   a  [        R                  $ U [        [        5      ;   d!  U [        [        5      R                  5       ;   a  [        R                  $ U [        R                  :X  a  [        R                  $ g)Nunknown)varsrP  valuesr   cvrW  nlprs  audiork  multi_modalrw  default)attribute_or_values    r8  get_trainer_domainTrainers.get_trainer_domain8  s    " 1T*5E5L5L5NN994$  2d7!!'!*::4$ "4!9##)68#,<<4"$$ $'9T&>((.(1%%%8#3#33###r7  r   N)
r   r   r   r   r   r  r   staticmethodr  r6  r   r7  r8  rw  rw  +  s&     G) r7  rw  c                   d   \ rS rSrSrSrSrSrSrSr	Sr
S	rS
rSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSr Sr!S r"S!r#S"r$S#r%S$r&S%r'S&r(S'r)S(r*S)r+S*r,S+r-S,r.S-r/S.r0S/r1S0r2S1r3S2r4S3r5S4r6S5r7S6r8S7r9S8r:S9r;S:r<S;r=S<r>S=r?S>r@S?rAS@rBSArCSBrDSCrESDrFSErGSFrHSGrISHrJSIrKSJrLSKrMSLrNSMrOSNrPSOrQSPrRSQrSSRrTSSrUSTrVSUrWSVrXgW)XPreprocessorsiO  aa  Names for different preprocessor.

    Holds the standard preprocessor name to use for identifying different preprocessor.
This should be used to register preprocessors.

    For a general preprocessor, just use the function name as preprocessor name such as
resize-image, random-crop
    For a model-specific preprocessor, use ${modelname}-${function}
z
load-imagezimage-denoise-preprocessorzimage-deblur-preprocessorz%object-detection-tinynas-preprocessorz&image-classification-mmcv-preprocessorz image-color-enhance-preprocessorz(image-instance-segmentation-preprocessorz%image-driving-perception-preprocessorz'image-portrait-enhancement-preprocessorz)image-quality_assessment-man-preprocessorz)image-quality_assessment-mos-preprocessorz video-summarization-preprocessorz%movie-scene-segmentation-preprocessorz(image-classification-bypass-preprocessorzobject-detection-scrfdzimage-sky-change-preprocessorzimage-demoire-preprocessorrQ  zocr-detectionz bad-image-detecting-preprocessorznerf-recon-acc-preprocessorznerf-recon-4k-preprocessorz&nerf-recon-vq-compression-preprocessorz*controllable-image-generation-preprocessorz!image-classification-preprocessorzsen-sim-tokenizerzcross-encoder-tokenizerzbert-seq-cls-tokenizerztext-gen-tokenizerztext2text-gen-preprocessorztext-gen-jieba-tokenizerz text2text-translate-preprocessorztoken-cls-tokenizerzner-tokenizerzthai-ner-tokenizerzviet-ner-tokenizerznli-tokenizerzsen-cls-tokenizerzdialog-intent-preprocessorzdialog-modeling-preprocessorz"dialog-state-tracking-preprocessorzsbert-token-cls-tokenizerzzero-shot-cls-tokenizerrZ  r[  r]  rA  zsequence-labeling-tokenizerz'word-segment-text-to-label-preprocessorzthai-wseg-tokenizerr>  rW  z#faq-question-answering-preprocessorr\  z%table-question-answering-preprocessorzre-tokenizerr_  r?  zmglm-summarizationzsentence-piecez#translation-evaluation-preprocessorrX  zdialog-use-preprocessorzsiamese-uie-preprocessorrc  rd  re  rg  zlinear-aec-fbankztext-to-tacotron-symbolszwav-to-listsz
wav-to-scpzkantts-data-preprocessorzofa-tasks-preprocessorzclip-preprocessorzmplug-tasks-preprocessorr^  zvldoc-preprocessorzhitea-tasks-preprocessorz'diffusion-image-generation-preprocessorzmplug-owl-preprocessorz/image-captioning-clip-interrogator-preprocessorzunifold-preprocessorr   N)Yr   r   r   r   r   
load_imageimage_denoise_preprocessorimage_deblur_preprocessor%object_detection_tinynas_preprocessor&image_classification_mmcv_preprocessor image_color_enhance_preprocessor(image_instance_segmentation_preprocessor%image_driving_perception_preprocessor'image_portrait_enhancement_preprocessor)image_quality_assessment_man_preprocessor)image_quality_assessment_mos_preprocessor video_summarization_preprocessor%movie_scene_segmentation_preprocessor(image_classification_bypass_preprocessorobject_detection_scrfdimage_sky_change_preprocessorimage_demoire_preprocessorr   r    bad_image_detecting_preprocessornerf_recon_acc_preprocessornerf_recon_4k_preprocessor&nerf_recon_vq_compression_preprocessor*controllable_image_generation_preprocessor!image_classification_preprocessorsen_sim_tokenizercross_encoder_tokenizerbert_seq_cls_tokenizertext_gen_tokenizertext2text_gen_preprocessortext_gen_jieba_tokenizer text2text_translate_preprocessortoken_cls_tokenizerner_tokenizerthai_ner_tokenizerviet_ner_tokenizernli_tokenizersen_cls_tokenizerdialog_intent_preprocessordialog_modeling_preprocessor"dialog_state_tracking_preprocessorsbert_token_cls_tokenizerzero_shot_cls_tokenizerr  r  r  rH  sequence_labeling_tokenizer'word_segment_text_to_label_preprocessorthai_wseg_tokenizerrE  r  #faq_question_answering_preprocessorr  %table_question_answering_preprocessorre_tokenizerr  rF  mglm_summarizationsentence_piecer  r  dialog_use_preprocessorsiamese_uie_preprocessorr  r  r  r  linear_aec_fbanktext_to_tacotron_symbolswav_to_lists
wav_to_scpkantts_data_preprocessorofa_tasks_preprocessorclip_preprocessormplug_tasks_preprocessorr  vldoc_preprocessorhitea_tasks_preprocessor'diffusion_image_generation_preprocessormplug_owl_preprocessor/image_captioning_clip_interrogator_preprocessorunifold_preprocessorr6  r   r7  r8  r  r  O  s    J!= ;,S)-U*'I$/Y,,S).W+0[-0[-'I$,S)/Y,5$C!!='O#M'I$"?!=-U*1].(K% ,75-!=9'I$/#M--#M+!=#A )M& ;73%N-!L"?.W+/I'O*O'!=,S)!L3--%NB+79)M&&G#(K%,S) *9!LJ9 6+9!L-9.W+56g3 2r7  r  c                       \ rS rSrSrSrSrSrSrSr	Sr
S	rS
rSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrSr Sr!S r"g!)"Metricsi  z!Names for different metrics.
    accuracymAPzaudio-noise-metricpplbleuzimage-denoise-metricz video-frame-interpolation-metriczvideo-super-resolution-metriczimage-ins-seg-coco-metriczseq-cls-metriczloss-metricztoken-cls-metricztext-gen-metriczprediction-saving-wrapperzimage-color-enhance-metricz!image-portrait-enhancement-metriczvideo-summarization-metriczmovie-scene-segmentation-metriczimage-inpainting-metricnedinbatch_recallz*referring-video-object-segmentation-metriczvideo-stabilization-metricz#image-quality-assessment-mos-metricz+image-quality-assessment-degradation-metricztext-ranking-metriczimage-colorization-metriczocr-recognition-metricztranslation-evaluation-metricr   N)#r   r   r   r   r   r  multi_average_precisionaudio_noise_metricPPLBLEUimage_denoise_metric video_frame_interpolation_metricvideo_super_resolution_metricimage_ins_seg_coco_metricseq_cls_metricloss_metrictoken_cls_metrictext_gen_metricprediction_saving_wrapperimage_color_enhance_metric!image_portrait_enhancement_metricvideo_summarization_metricmovie_scene_segmentation_metricimage_inpainting_metricNEDr  *referring_video_object_segmentation_metricvideo_stabilization_metric#image_quality_assessment_mos_metric+image_quality_assessment_degradation_metrictext_ranking_metricimage_colorization_metricocr_recognition_metrictranslation_evaluation_metricr6  r   r7  r8  r  r    s     H#-
C D 2'I$$C! !<%NK)'O ;!=(K%!=&G#7
C%N1].!=*O'2_// ;5$C!r7  r  c                        \ rS rSrSrSrSrSrg)
Optimizersi  zNames for different OPTIMIZER.

Holds the standard optimizer name to use for identifying different optimizer.
This should be used to register optimizer.
	optimizerSGDr   N)r   r   r   r   r   r  r  r6  r   r7  r8  r  r    s     G
Cr7  r  c                   l    \ rS rSrSrSrSrSrSrSr	Sr
S	rS
rSrSrSrSrSrSrSrSrSrSrSrSrSrSrg)Hooksi  z@Names for different hooks.

All kinds of hooks are defined here
LrSchedulerHookPlateauLrSchedulerHookNoneLrSchedulerHookOptimizerHookTorchAMPOptimizerHookApexAMPOptimizerHookNoneOptimizerHookCheckpointHookBestCkptSaverHookLoadCheckpointHookTextLoggerHookTensorboardHookIterTimerHookEvaluationHookSparsityHookClipClampLogitScaleHookEarlyStopHookDeepspeedHookMegatronHookDDPHook	SwiftHookr   N)r   r   r   r   r   r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   r  r  r  r  r  r  r6  r   r7  r8  r  r    s    
 (O5/ $M31+ &N+- &N'O#M%N "L 8 $M#M!LGIr7  r  c                   $    \ rS rSrSrSrSrSrSrg)LR_Schedulersi-  z-learning rate scheduler is defined here

    LinearWarmupConstantWarmupExponentialWarmupr   N)	r   r   r   r   r   r	  r
  r  r6  r   r7  r8  r  r  -  s     "L%N+r7  r  c                   (    \ rS rSrSrSrSrSrSrSr	g)	CustomDatasetsi6  z"Names for different datasets.
    PairedDatasetSiddDatasetGoproDatasetRedsDatasetr   N)
r   r   r   r   r   r  r  r  r  r6  r   r7  r8  r  r  6  s    #MK!LKr7  r  N)modelscope.utils.constantr   r   objectr   r:  rK  rR  r  rH  r  r  r  rC  r  r  rD  r  competency_aware_translationr  r   r  rY  r  rm  rn  r}  image_object_detectionr  image_denoisingro  image_deblurringrp  rq  r   r  rB  rG  r  task_oriented_conversationr  r  table_question_answeringr  r  r  r  r  r0  r?  image_portrait_stylizationrs  r   rt  r   rB  ru  rv  rE  rF  rw  r  r  r  r1  r2  r6  r3  r@  video_embeddingrz  r7  rD  r   r   r|  r  r   r  r  face_livenessr  r  r  r  r   r  r8  text_video_retrievalr9  image_color_enhancementr  r  r  r  r  image_segmentationr  image_driving_perceptionr  r  r  r   r  r  r  r  r  r  r  r  r  r  r   r  r  r  r  r   r  r  r   r  r   r   r  r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r   r  r  r  r   r   r  r   r   r   rA   r  r   r   r   r   r   r  r   r2  r   r   r  r  r   r   r   DEFAULT_MODEL_FOR_PIPELINErP  rW  rk  rs  rw  r  r  r  r  r  r  r   r7  r8  <module>r"     s   3*V *D	D 	D(F (.I( I(X
o	!!57o
 
//ACo 
//FHo 
  9;o 
933LNo 
68o" 
""''@B%o( 
""57+o. 
  ""571o4 
"";=7o: 
&&  "JK=o> 
	44;=?oB 
II	EFCoD 
""''@GoL 
Y779;MoP 
i99>@QoT 
I55>@UoX 
  9#=#=#F#HYo\ 
I33?A]o` 
Y33@Baod 
)11:<eoh 
	 = = K Miol 
  %%:<oor 
''@Buox 
I55KMyo| 
""''@BoB 
$$y'@'@'G'ICoF 
)"A"A"H"JGoJ 
""00+-MoP 
++0013SoV 
))..&(Yo\ 
,,1102_ob 
$$24eoh 
Y77CEiol 
68oor 
$$##8:uox 
)11JLyo| 
  @BoB 
$$))BDEoH 
$$=?KoN 
!!&&46QoT 
##i&I&I&H&JUoX 
OOi))+JKYoZ 
y;;GI[o^ 
y;;DF_ob 
Y77MOcof 
)11:<goj 
933<>kon 
)"A"A"M"Ooor 
**//Ouoz 
  %%EG}o@ 
##((9;CoF 
""''>@IoL 
I<<;=MoP 
!!&&57SoV 
!!I$E$E$B$DWoZ 
i99JL[o^ 
i99IK_ob 
  :<eoh 
933EGiol 
)11=?mop 
!!:<sov 
)44=?woz 
Y77DF{o~ 
'',,68AoD 
$$));=GoJ 
i99MOKoN 
!!&&13QoT 
%%**13WoZ 
!E!E!J!L[o^ 
!!""02aod 
933?Aeoh 
yCCCEiol 
y;;@Bmop 
**68sov 
""6679yo| 
  %%46oB 
!!&&=?EoH 
!!&&79KoN 
""''/1QoT 
  %%9;WoZ 
##((02]o` 
'',,46cof 
&&++LNiol 
!?!?!C!Emop 
)"A"A"E"Gqot 
  9#C#C#H#Juox 
I55:<yo| 
$$)).0oB 
%%**46EoH 
##((46KoN 
,,@!BOoT 
  --3#5UoZ 
!!;=[o` 
I55:<aod 
  ((A#Ceoj 
933@Bkon 
&&..;)=oot 
i99IKuox 
""**5%7yo~ 
""**A%CoD 
i99DFEoH 
Y77:<IoL 
!?!?!L!NMoP 
'',,KMSoV 
Y7779WoZ 
	 = = F H[o^ 
!?!?!K!M_ob 
!!))2$4coh 
##++0&2ion 
i99>@oor 
""**9%;sox 
	--8:yo| 
##++3&5}oB	 
//1GHC	oD	 
!?!?!E!GE	oH	 
--55<0>I	oN	 
	 = = K MO	oR	 
I558:S	oV	 
  ((F#HW	o\	 
##++2&4]	ob	 
%%--<(>c	oh	 
++33<.>i	on	 
!!))4$6o	ot	 
!?!?!I!Ku	ox	 
i99CEy	o|	 
  ;=}	oB
 
&&..@)BC
oH
 
..66?1AI
oN
 
!!))G$IO
oT
 
	 B B B DU
oX
 
	 = = J LY
o\
 
933KM]
o`
 
)11GIa
od
 
##++<&>e
oj
 
!?!?!H!Jk
on
 
	--FH	**22A-C 
##++@&B 
//:<	  9#C#C#K#M	933?A	i99:<	!?!?!?!A	##++9&; 
**22/-1Yo dJ J,F& F,* *,F ,!z;(:M !Hf2F f2R<Df <D~	 	'F 'T,F , V  r7  