o rq iE@sddlmZddlmZddlmZddlmZmZddl m Z ddl m Z m Z ddlmZe Ze dZeZe d ZeeZeeD]Zed ejkrfedd krfeeeejed ed f<qG ddededefddZddedefddZ ddededefddZ d S))Models) ConfigDict)Tasks) INDEX_KEYLazyImportModule) get_logger)Registrybuild_from_cfg)get_task_by_subtask_namemodelsZheads BACKBONESNcfg task_name default_argsc Csrz t|t||d}W|Sty8}z!t|\}}|dur!t|||d<t|t||d}WYd}~|Sd}~ww)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. Z group_keyrNtype)r MODELSKeyErrorr )rrrmodeleZ parent_taskZtask_model_typer`/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/modelscope/models/builder.py build_models    rc Cs|dds |dd}n|dd}z t|ttj|d}W|StyEtd|j dt j |d<||d<t|ttj|d}Y|Sw) z build backbone given backbone config dict Args: cfg (:obj:`ConfigDict`): config dict for backbone object. default_args (dict, optional): Default initialization arguments. Z init_backboneF model_dirNrz The backbone zR is not registered in modelscope, try to import the backbone from hf transformers.r) getpopr r rbackbonerloggerwarningrrZ transformers)rrrrrrrbuild_backbone2s2     r!cCst|t||dS)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)rrrrrr build_headUs r#)NN)N)!Zmodelscope.metainforZmodelscope.utils.configrZmodelscope.utils.constantrZmodelscope.utils.import_utilsrrZmodelscope.utils.loggerrZmodelscope.utils.registryrr Zmodelscope.utils.task_utilsr rrr r"Z get_ast_indexmoduleslistkeysZ module_indexrnameupperstrdictrr!r#rrrrsD       $