ujEddlmZddlmZddlmZddlmZmZddl m Z ddl m Z m Z ddlmZe Ze dZeZe d ZejeZeeD]PZed ejkr=edd kr1eeeejed ed f<Q ddededefdZddedefdZ ddededefdZ d S))Models) ConfigDict)Tasks) INDEX_KEYLazyImportModule) get_logger)Registrybuild_from_cfg)get_task_by_subtask_namemodelsheads BACKBONESNcfg task_name default_argsc t|t||}nW#t$rJ}t|\}}|t|||d<t|t||}Yd}~nd}~wwxYw|S)a build model given model config dict Args: cfg (:obj:`ConfigDict`): config dict for model object. task_name (str, optional): task name, refer to :obj:`Tasks` for more details default_args (dict, optional): Default initialization arguments.  group_keyrNtype)r MODELSKeyErrorr )rrrmodele parent_tasktask_model_types i/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/modelscope/models/builder.py build_modelrs K 9<III  K K K(@ 'J'J$ _  "1++ %F  ;\KKK K Ls A/AA**A/c|dds|dd}n|dd} t|ttj|}ni#t $r\td|j dtj |d<||d<t|ttj|}YnwxYw|S) z build backbone given backbone config dict Args: cfg (:obj:`ConfigDict`): config dict for backbone object. default_args (dict, optional): Default initialization arguments. init_backboneF model_dirNrz The backbone zR is not registered in modelscope, try to import the backbone from hf transformers.r) getpopr rrbackbonerloggerwarningrr transformers)rrr"rs rbuild_backboner)2s 77?E * */GGK.. GGK.. '  n% '''  ' ' '  xCH x x x   )F $K  n% ''' ' Ls"A((A#C Cc2t|t||S)a build head given config dict Args: cfg (:obj:`ConfigDict`): config dict for head object. task_name (str, optional): task name, refer to :obj:`Tasks` for more details default_args (dict, optional): Default initialization arguments. r)r HEADS)rrrs r build_headr,Us(  Uil D D DD)NN)N)!modelscope.metainformodelscope.utils.configrmodelscope.utils.constantrmodelscope.utils.import_utilsrrmodelscope.utils.loggerrmodelscope.utils.registryr r modelscope.utils.task_utilsr r&rrr+ get_ast_indexmoduleslistkeys module_indexr%nameupperstrdictrr)r,r-rr?s&&&&&&......++++++EEEEEEEE......>>>>>>>>@@@@@@  (     (  ( * *9 5D((;;LA%.((\!_ -K-K%,\%: ""$$l1oa" # "&%)Z"6   $    H!%$( D DJ D D! D D D D D Dr-