o ॵi3 @s ddlZddlmZddlmZmZmZddlm Z ddl m Z m Z ddl mZddlmZddlmZddlmZmZmZd d lmZeZe je jefd ed ed eedeedeef ddZefdedeefddZdeeeeffddZddZ ddZ!dS)N)ListOptionalUnion) HTTPError)LicensesModelVisibility)model_file_download)snapshot_download)Config)DEFAULT_MODEL_REVISION ConfigFields ModelFile) get_loggermodel_id chinese_name visibilitylicenserevisioncCsld}z |j||dWn tyd}Ynw|r#td|ddS|j||||dtd|ddS)NT)rrFzmodel z already exists, skip creation.)rrrrz successfully created.) get_modelrprint create_model)apirrrrrexistsrH/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/utils/hub.pycreate_model_if_not_exists" rmodel_id_or_pathcCsVtj|st|tj|d}ntj|rtj|tj}ntj|r&|}t |S)z Read config from hub or local path Args: model_id_or_path (str): Model repo name or local directory path. revision: revision of the model when getting from the hub Return: config (:obj:`Config`): config object )r) ospathrrr CONFIGURATIONisdirjoinisfiler from_file)rr local_pathrrr read_config.s    r&modelcCs2t|trt|st|}|Sdd|D}|S)NcSs"g|] }t|s t|n|qSr)osprr ).0mrrr Hszauto_load..) isinstancestrr(rr )r'rrr auto_loadCs  r.c CszRt|tj}t|d}t|r:t|}t|jdr$|jj j WSt|jdr5t|jds5|jj WS|jj WSt|rPt|}t|drM|j WSdWSWdSt ym}zt d|WYd}~dSd}~ww)aGet the model type from the configuration. This method will try to get the model type from 'model.backbone.type', 'model.type' or 'model.model_type' field in the configuration.json file. If this file does not exist, the method will try to get the 'model_type' field from the config.json. Args: model_dir: The local model dir to use. @return: The model type string, returns None if nothing is found. config.jsonbackbone model_typetypeNz%parse config file failed with error: )r(r"r r r#r r$hasattrr'r0r2r1 Exceptionloggererror) model_dirconfiguration_file config_filecfgerrrget_model_typeOs,           r<c Csddl}ddl}d}|j|tj}|j|rz'parse_label_mapping..label2idid2labelcSi|]\}}||qSrrr)idlabelrrrr@rAcSrDrrrErrrr@rAr/cSrDrrrErrrr@rAcSsi|] \}}|t|qSr)int)r)rGrFrrrr@s)jsonrrr"r LABEL_MAPPINGropenloaditemsr r r$r3r r'rBrC preprocessor) r7rIrrB label_pathf label_mapping config_pathconfigrCrrrparse_label_mappingnsR            rT)"ros.pathrr(typingrrrrequestsrmodelscope.hub.constantsrrmodelscope.hub.file_downloadr modelscope.hub.snapshot_downloadr modelscope.utils.configr modelscope.utils.constantr r r r5rPUBLIC APACHE_V2r-rHrr&r.r<rTrrrrs@