o
    )j                     @   s	  d dl mZmZ G dd deZG dd deZG dd deZG dd	 d	eZi ejejd
fej	ej
dfej	ej	dfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfi ejejdfejejdfejej dfej!ej!dfej"ej"dfej#ej#dfej$ejdfej%ej%dfej&ej&d fej'ej(d!fej)ej)d"fej*ej+d#fej,ej,d$fej-ej-d%fej.ej.d&fej/ej/d'fej0ej0d(fi ej1ej1d)fej2ej3d*fej4ej4d+fej5ej5d,fej6ej6d-fej7ej7d.fej8ej8d/fej9ej9d0fej:ej:d1fej;ej;d2fej<ej<d3fej=ej=d4fej>ej>d5fej?ej?d6fej@ej@d7fejAejAd8fejBejBd9fi ejCejCd:fejDejDd;fejEejFd<fejGejGd=fejHejHd>fejIejId?fejJejJd@fejKejKdAfejLejLdBfejMejMdCfejNejOdDfejPejQdEfejRejRdFfejSejSdGfejTejTdHfejUejUdIfejVejVdJfi ejWejWdKfejXejYdLfejZej[dMfej\ej\dNfej]ej^dOfej_ej_dPfej`ejadQfejbejcdRfejdejddSfejeejedTfejfejfdUfejgejgdVfejhejhdWfejiejidXfejjejjdYfejkejkdZfejlejld[fi ejmejmd\fejnejnd]fejoejod^fejpejpd_fejqejqd`fejrejrdafejsejtdbfejejudcfejvejvddfejwejwdefejxejxdffejyejydgfejzejzdhfej{ej{difej|ej|djfej}ej}dkfej~ej~dlfi ejejdmfejejdnfejejdofejejdpfejejdqfejejdrfejejdsfejejdtfejejdufejejdvfejejdwfejejdxfejejdyfejejdzfejejd{fejejd|fejejd}fi ejejd~fejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfejejdfi
ZG dd deZG dd deZG dd deZG dd deZG dd deeeeZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZdS )    )FieldsTasksc                   @   s  e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"Z$d#Z%d$Z&d%Z'd&Z(d'Z)d(Z*d)Z+d*Z,d+Z-d,Z.d-Z/d.Z0d/Z1d0Z2d1Z3d2Z4d3Z5d4Z6d5Z7d6Z8d7Z9d8Z:d9Z;d:Z<d;Z=d<Z>d=Z?d>Z@d?ZAd@ZBdAZCdBZDdCZEdDZFdEZGdFZHdGZIdHZJdIZKdJZLdKZMdLZNdMZOdNZPdOZQdPZRdQZSdRZTdSZUdTZVdUZWdVZXdWZYdXZZdYZ[dZZ\d[Z]d\Z^d]Z_d^Z`d_Zad`ZbdaZcdbZddcZeddZfdeZgdfZhdgZidhZjdiZkdjZldkZmdlZndmZodnZpdoZqdpZrdqZsdrZtdsZudtZvduZwdvZxdwZydxZzdyZ{dzZ|d{Z}d|Z~d}Zd~ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdS )Modelsz Names 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	detectionZMaskScoringzimage-restorationzrealtime-object-detectionzrealtime-video-object-detectionscrfddepeZClassificationModelZEasyRobustModelbnextyolopv2nafnetcsrnetadaint
deeplpfnetrrdbcascade_mask_rcnn_swinmaskdino_swingpenzproduct-retrieval-embeddingzbody-2d-keypointszbody-3d-keypointsZhdformerZHRNetCrowdCountingzface-2d-keypointszstar-68ldk-detectionzswinL-panoptic-segmentationzr50-panoptic-segmentationZpassvitbZ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ddpmZOCRRecognitionZOCRDetectionZ	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llama2Z	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-symmetryN)__name__
__module____qualname____doc__tinynas_detectiontinynas_damoyolor   Zmask_scoringZimage_restorationZrealtime_object_detectionrealtime_video_object_detectionr   r	   Zclassification_modelZeasyrobust_modelr
   r   r   r   r   r   r   r   r   r   product_retrieval_embeddingbody_2d_keypointsbody_3d_keypointsZbody_3d_keypoints_hdformercrowd_countingface_2d_keypointsZstar_68ldk_detectionZpanoptic_segmentationZr50_panoptic_segmentationimage_reid_personimage_inpaintingimage_paintbyexamplevideo_summarizationvideo_panoptic_segmentationvideo_instance_segmentation#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_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_deinterlaceZ!quadtree_attention_image_matchingZ"loftr_image_local_feature_matchingZlightglue_image_matchingZvision_middlewarer0   video_stabilizationZreal_basicvsrZrcp_sceneflow_estimationZimage_casmvs_depth_estimationZ image_geomvsnet_depth_estimationZvop_retrieval_modelZvop_retrieval_model_ser2   Zimage_probing_modelr3   image_face_fusioncontent_checkopen_vocabulary_detection_vildr6   Zmsrresnet_liter7   r8   ocr_recognitionocr_detectionlineless_table_recognitionimage_quality_assessment_mosimage_quality_assessment_man$image_quality_assessment_degradationr<   nerf_recon_accnerf_recon_4knerf_recon_vq_compressionsurface_recon_commonZbts_depth_estimationvision_efficient_tuningbad_image_detectingcontrollable_image_generationrD   rE    pedestrian_attribute_recognitionimage_try_onhuman_image_generationimage_view_transformimage_control_3d_portraitrJ   rK    self_supervised_depth_completionrM   ZpalmrN   rO   rP   translationrR   Z	space_dstZspace_intentZspace_modelingZ
space_T_enZ
space_T_cnZtcrfZtoken_classification_for_nerZ	tcrf_wsegZtransformer_softmaxZlcrfZ	lcrf_wsegZgcnncrfrU   rV   rW   Zgpt_moeZgpt_neorX   Zbert_for_dsZponet_for_dsrY   rZ   r[   r\   r]   r^   r_   r`   Zmegatron_bertZuseZfid_plugZfid_T5rb   Zxlm_robertarc   Zplug_mentalrd   re   rf   rg   Z
chatglm_6bZchatglm2_6bZqwen_7bZsambert_hifiganrh   ri   rj   rk   rl   rm   rn   ro   	kws_kwsbpZgeneric_asrZ	wenet_asrZgeneric_itnZgeneric_puncZ
generic_svZtdnn_svZecapa_tdnn_svZcampplus_svZeres2net_svZeres2netv2_svZ	resnet_svZ
res2net_svZeres2net_aug_svZscl_sdZscl_sd_xvectorZcampplus_lreZeres2net_lreZcluster_backendZrdino_tdnn_svZsdpn_svZ
generic_lmaudio_quantizationZlaura_codecrr   rs   rt   ru   ZclipZgemmZrlegrw   Z	diffusionZmulti_stage_diffusionZvideo_synthesisZteamZ
video_clipZprostry   rz   r{   r|   efficient_diffusion_tuningcones2_inferenceZ	mplug_owlZclip_interrogatorstable_diffusionZstable_diffusion_xlr   text_to_360panorama_imageZimage_to_video_modelZvideo_to_video_modelr   Zunifold_symmetry r   r   T/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/metainfo.pyr      s   r   c                   @   s,   e Zd ZdZdZdZdZdZdZdZ	dZ
d	S )

TaskModelstext-classificationtoken-classificationinformation-extraction	fill-maskfeature-extractiontext-generationtext-rankingzmachine-reading-comprehensionN)r   r   r   text_classificationtoken_classificationinformation_extraction	fill_maskfeature_extractiontext_generationtext_rankingZmachine_reading_comprehensionr   r   r   r   r     s    r   c                   @   8   e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdS )Headsr   r   zbert-mlmzroberta-mlmzxlm-roberta-mlmr   r   r   r   rT   rS   N)r   r   r   r   r   Zbert_mlmZroberta_mlmZxlm_roberta_mlmr   r   r   r   Zlstm_crfZtransformer_crfr   r   r   r   r     s    r   c                   @   s  e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"Z$d"Z%d#Z&d$Z'd%Z(d&Z)d'Z*d(Z+d)Z,d*Z-d+Z.d,Z/d-Z0d.Z1d/Z2d0Z3d1Z4d2Z5d3Z6d4Z7d5Z8d6Z9d7Z:d8Z;d9Z<d:Z=d;Z>d<Z?d=Z@d>ZAd?ZBd@ZCdAZDdBZEdCZFdDZGdEZFdFZHdGZIdHZJdIZKdJZLdKZMdLZNdMZOd.Z1dNZPdOZQdPZRdQZSdRZTdSZUdTZVdUZWdVZXdWZYdXZZdYZ[dZZ\d[Z]d\Z^d]Z_d^Z`d_Zad`ZbdaZcdbZddcZeddZfdeZgdfZhdgZidhZjdiZkdjZldkZmdlZndmZodnZpdoZqdpZrdqZsdrZtdsZudtZvduZwdvZxdwZydxZzdyZ{dzZ|d{Z}d|Z~d}Zd~ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"Z$d#Z%d$Z&d%Z'd&Z(d'Z)d(Z*d)Z+d*Z,d+Z-d,Z.d-Z/d.Z0d/Z1d0Z2d1Z3d2Z4d3Z5d4Z6d5S (6  	Pipelinesa   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-segmentationZ!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%   r&   r'   r(   zflow-based-body-reshaping#referring-video-object-segmentationr)   r*   r+   r,   zvision-middleware-multi-taskr0   r-   r.   r/   zimage-matchingzimage-matching-fastr1   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-detectionr4   r5   z ddpm-image-semantic-segmentationzvideo-colorizationzmdm-motion-generationzmobile-image-super-resolutionzm2fp-image-human-parsingzobject-detection-3d-deper=   r>   r?   r@   rB   rC   zfast-instance-segmentationr9   r:   r;   rA   zimage-bts-depth-estimationzimage-depth-estimation-marigoldz/resnet50_pedestrian-attribute-recognition_imager   rF   rG   zhuman3d-renderzhuman3d-animationrH   rI   zrife-video-frame-interpolationrK   zimage-to-3drL   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-ponetrQ   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-embeddingr   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-evaluationra   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-16krh   ri   speech_dfsmn_ans_psm_48k_causalrk   speech-separationrp   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-inferencerq   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~   rx   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-pipelineN(7  r   r   r   r   Zpipeline_templateportrait_mattinguniversal_mattingimage_denoiseimage_deblurimage_editingZ!freeu_stable_diffusion_text2imageperson_image_cartoonr   table_recognitionr   license_plate_detectioncard_detection_correctionaction_recognitionZanimal_recognitionZgeneral_recognitioncmdssl_video_embeddingZhicossl_video_embeddingr   r   hand_2d_keypointshuman_detectionZtbs_detectionobject_detectionZabnormal_object_detectionr   Zsalient_detectionZsalient_boudary_detectionZcamouflaged_detectionZimage_demoireimage_classificationface_detectionface_liveness_irZface_liveness_rgbZface_liveness_xccard_detectionZulfd_face_detectionZtinymog_face_detectionfacial_expression_recognitionZfacial_landmark_confidenceZfacial_68ldk_detectionface_attribute_recognitionZretina_face_detectionmog_face_detectionZmtcnn_face_detectionlive_categoryZgeneral_image_classificationdaily_image_classificationZ(nextvit_small_daily_image_classificationZ*convnext_base_image_classification_garbageZ bnext_small_image_classification'yolopv2_image_driving_percetion_bdd100kZcommon_image_classificationimage_color_enhanceZadaint_image_color_enhanceZdeeplpf_image_color_enhancevirtual_try_onimage_colorizationimage_style_transferimage_super_resolutionZimage_super_resolution_pasdimage_debandingface_image_generationr   r   face_recognitionZface_recognition_oodface_quality_assessmentZface_recognition_onnx_irZface_recognition_onnx_fmZarc_face_recognitionZmask_face_recognitionr   image_instance_segmentationZmaskdino_instance_segmentationZimage2image_translationvideo_categoryr   image_portrait_enhancementimage_to_image_generationimage_object_detection_autoZhand_detectionskin_retouchingZface_reconstructionZhead_reconstructionZtext_to_headZtinynas_classificationZeasyrobust_classificationr   r   action_detectionvideo_single_object_trackingZ'video_single_object_tracking_procontextvideo_multi_object_trackingZimage_panoptic_segmentationr   r   Zimage_semantic_segmentationimage_depth_estimationimage_normal_estimationindoor_layout_estimationimage_local_feature_matchingvideo_depth_estimationpanorama_depth_estimationZpanorama_depth_estimation_s2netdense_optical_flow_estimationr   r   r   Zimage_inpainting_sdv2r   movie_scene_segmentationr   r   r   Zpst_action_recognitionr   r   r   r   r   r   r   r   r   r   Zvision_middleware_multi_taskr0   r   r   r   image_matchingZimage_matching_fastr   video_super_resolutionZpointcloud_sceneflow_estimation!image_multi_view_depth_estimationr   r   Zvop_retrievalZvop_retrieval_seddcolor_image_colorizationZimage_structured_model_probingimage_fewshot_detectionr   r   Z ddpm_image_semantic_segmentationvideo_colorizationZmotion_generattionZmobile_image_super_resolutionZimage_human_parsingobject_detection_3d_deper   r   r   r   r   r   Zfast_instance_segmentationr   r   r   r   Zimage_bts_depth_estimationZimage_depth_estimation_marigoldr   r   r   r   human3d_renderhuman3d_animationr   r   Zrife_video_frame_interpolationrK   Zimage_to_3dr   r   Zautomatic_post_editingZtranslation_quality_estimationZdomain_classificationsentence_similarityword_segmentationZmultilingual_word_segmentationZword_segmentation_thaipart_of_speechnamed_entity_recognitionZnamed_entity_recognition_thaiZnamed_entity_recognition_vietr   Zfid_dialogueZtext2text_generationZsentiment_analysissentiment_classificationr   r   fill_mask_ponetcsanmt_translationcanmt_translationZinteractive_translationr   Zdialog_intent_predictiondialog_modelingdialog_state_trackingzero_shot_classificationtext_error_correctionword_alignmentZplug_generationZgpt3_generationZpolylm_text_generationZgpt_moe_generationfaq_question_answeringconversational_text_to_sql!table_question_answering_pipelinesentence_embeddingr   mgeo_rankingrelation_extractiondocument_segmentationZextractive_summarizationr   Zmglm_text_summarizationZcodegeex_code_translationZcodegeex_code_generationZglm130b_text_generationr   r   r   r   translation_evaluationZuser_satisfaction_estimationsiamese_uie"document_grounded_dialog_retrievaldocument_grounded_dialog_rerank!document_grounded_dialog_generater   %machine_reading_comprehension_for_nerr   Zsambert_hifigan_ttsZspeech_dfsmn_aec_psm_16krh   ri   r   rk   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lm_inferenceZspeech_timestamp_inferencer   Zaudio_quantization_inferenceZlaura_codec_tts_inferenceZ!speech_super_resolution_inferenceZvoice_conversionimage_captioningmulti_modal_embedding generative_multi_modal_embeddingvisual_question_answeringZvisual_groundingZvisual_entailmentmulti_modal_similaritytext_to_image_synthesisvideo_multi_modal_embeddingprost_text_video_retrievalr   Zimage_text_retrievalZofa_ocr_recognitionZofa_asrZ
ofa_sudokuZofa_text2sqlvideo_captioningvideo_question_answeringZdiffusers_stable_diffusionr   document_vl_embeddingZchinese_stable_diffusionr   text_to_video_synthesisZ"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protein_structureZfunasr_pipeliner   r   r   r   r   *  sn   	r   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                   @   T   e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdZdZdZdZdS )
CVTrainerszimage-instance-segmentationzimage-portrait-enhancementzvideo-summarizationzmovie-scene-segmentationzface-detection-scrfdzcard-detection-scrfdzimage-inpaintingr   zimage-classification-teamr   r   ocr-recognitionzocr-detection-dbr=   r>   zaction-detectionrA   rL   N)r   r   r   r  r  r   r#  Zface_detection_scrfdZcard_detection_scrfdr   r   Zimage_classification_teamr   r(  r   Zocr_detection_dbr   r   r  r   r   r   r   r   r   rU    &    rU  c                   @   rT  )NLPTrainersz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-trainerN)r   r   r   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gpt3_trainerZfaq_question_answering_trainerZ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   r   r   r   rX    rW  rX  c                   @   r   )MultiModalTrainersrv   ru   rw   zmgeo-ranking-trainerr}   r   zlora-diffusionzlora-diffusion-xlzdreambooth-diffusionzcustom-diffusionr~   N)r   r   r   Zclip_multi_modal_embeddingru   rw   Zmgeo_ranking_trainerr   r   Zlora_diffusionZlora_diffusion_xlZdreambooth_diffusionZcustom_diffusionr   r   r   r   r   rY    s    rY  c                   @   s$   e Zd ZdZdZdZdZdZdZdS )AudioTrainersrh   rk   rm   zspeech-kantts-trainerzspeech-asr-trainerr   N)	r   r   r   rh   rk   rm   Zspeech_kantts_trainerZspeech_asr_trainerrG  r   r   r   r   rZ  "  s    rZ  c                   @   s$   e Zd ZdZdZdZedd ZdS )Trainersa2   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.
    Ztrainerr   c                 C   s   | t tv s| t t v rtjS | t tv s| t t v r"tjS | t tv s0| t t v r3tjS | t t	v sA| t t	 v rDtj
S | tjkrLtjS dS )Nunknown)varsrU  valuesr   ZcvrX  ZnlprZ  ZaudiorY  Zmulti_modalr[  default)Zattribute_or_valuer   r   r   get_trainer_domain8  sF   
zTrainers.get_trainer_domainN)r   r   r   r   r_  r   staticmethodr`  r   r   r   r   r[  +  s    	r[  c                   @   s\  e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"Z$d#Z%d$Z&d%Z'd&Z(d'Z)d(Z*d)Z+d*Z,d+Z-d,Z.d-Z/d.Z0d/Z1d0Z2d1Z3d2Z4d3Z5d4Z6d5Z7d6Z8d7Z9d8Z:d9Z;d:Z<d;Z=d<Z>d=Z?d>Z@d?ZAd@ZBdAZCdBZDdCZEdDZFdEZGdFZHdGZIdHZJdIZKdJZLdKZMdLZNdMZOdNZPdOZQdPZRdQZSdRZTdSZUdTZVdUS )VPreprocessorsaz   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-preprocessorrV  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-tokenizerr   r   r   r   zsequence-labeling-tokenizerz'word-segment-text-to-label-preprocessorzthai-wseg-tokenizerr   r   z#faq-question-answering-preprocessorr   z%table-question-answering-preprocessorzre-tokenizerr   r   zmglm-summarizationzsentence-piecez#translation-evaluation-preprocessorr   zdialog-use-preprocessorzsiamese-uie-preprocessorr   r   r   r   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-preprocessorN)Wr   r   r   r   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_preprocessorr   r   Z 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_tokenizerr8  r9  r=  r   Zsequence_labeling_tokenizerZ'word_segment_text_to_label_preprocessorZthai_wseg_tokenizerr   r2  Z#faq_question_answering_preprocessorr;  Z%table_question_answering_preprocessorZre_tokenizerr@  r   Zmglm_summarizationZsentence_piecerA  r4  Zdialog_use_preprocessorZsiamese_uie_preprocessorrC  rD  rE  rF  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   r   r   r   rb  O  s    rb  c                   @   s   e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dS ) Metricsz" Names for different metrics.
    accuracyZmAPzaudio-noise-metricZpplZ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-metricZ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-metricN)!r   r   r   r   rd  Zmulti_average_precisionZaudio_noise_metricZPPLZ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_metricZNEDre  Z*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   r   r   r   rc    s>    rc  c                   @   s   e Zd ZdZdZdZdS )
Optimizersz Names for different OPTIMIZER.

        Holds the standard optimizer name to use for identifying different optimizer.
        This should be used to register optimizer.
    Z	optimizerSGDN)r   r   r   r   r_  rg  r   r   r   r   rf    s    rf  c                   @   sd   e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdZdZdZdZdZdZdZdZdS )HookszM 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	SwiftHookN)r   r   r   r   ri  rj  rk  rl  rm  rn  ro  rp  rq  rr  rs  rt  ru  rv  rw  rx  ry  rz  r{  r|  r}  r   r   r   r   rh    s.    rh  c                   @   s   e Zd ZdZdZdZdZdS )LR_Schedulersz-learning rate scheduler is defined here

    LinearWarmupConstantWarmupExponentialWarmupN)r   r   r   r   r  r  r  r   r   r   r   r~  -  s
    r~  c                   @   s    e Zd ZdZdZdZdZdZdS )CustomDatasetsz# Names for different datasets.
    PairedDatasetSiddDatasetGoproDatasetRedsDatasetN)r   r   r   r   r  r  r  r  r   r   r   r   r  6  s    r  N)Zmodelscope.utils.constantr   r   objectr   r   r   r   r=  r   r>  r.  r/  r   r0  r?  r   r-  Zcompetency_aware_translationr4  r   r3  r   r1  r   r   r   Zimage_object_detectionr   Zimage_denoisingr   Zimage_deblurringr   r   r   r%  r   r   r7  Ztask_oriented_conversationr5  r6  Ztable_question_answeringr<  rE  rD  rC  r8  rH  rP  Zimage_portrait_stylizationr   r   r   r   rR  r   r   r   r   r   r  r  r  rI  rJ  rL  rK  rQ  Zvideo_embeddingr   rM  rS  r   r   r   r  r   r  r  Zface_livenessr  r  r  r  r   r  rN  Ztext_video_retrievalrO  Zimage_color_enhancementr
  r  r  r'  r)  Zimage_segmentationr  Z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   rA  r   r  r&  r(  r   r   r$  r   r   r   r7   r*  r   r   r   r   r   rB  r   r   r   r   r+  r,  r   r   r   ZDEFAULT_MODEL_FOR_PIPELINErU  rX  rY  rZ  r[  rb  rc  rf  rh  r~  r  r   r   r   r   <module>   s      N	!"&(*,.02469<>ACEHKNQTVY\^adgjlmoqsuwy}       	                       #  &  (  +  .  0  3  5  7  9  <  ?  B  E  H  K  N  Q  T  W  Y  [  ]  _  b  e  h  k  n  q  s  v  x  {  }                   
                                         #    $    &    )    +    -    0    3    6    9    <    >    @    C    F    I    L    N    P    R    T    W    Y    t	$i?*	