ujc ddlZddlmZmZddlmZmZddlmZddl m Z m Z ddl m Z ddlmZdd lmZeZGd d eZdS) N)ABCabstractmethod)DictUnion)Model)Config ConfigDict) ModelFile) get_logger)build_exportercdeZdZdZd dZedeeeffdZ e d defdZ dS) ExporterzNExporter base class to output model to onnx, torch_script, graphdef, etc. Nc||_dSN)model)selfrs i/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/modelscope/exporters/base.py__init__zExporter.__init__s  rc t|trtj|}t |dsJ|j}t jtj |tj }|j }t |dr|j}|j}t |drt |ds |j|_t%d|ji}t |dr||j||d< t+|||}n.#t,$r!} t-d|jd|d | d } ~ wwxYw|S) aXBuild the Exporter instance. Args: model: A Model instance or a model id or a model dir, the configuration.json file besides to which will be used to create the exporter instance. kwargs: Extra kwargs used to create the Exporter instance. Returns: The Exporter instance model_dir group_key model_typetypeexportrzThe exporting of model 'z' with task: 'z' is not supported currently.N) isinstancestrrfrom_pretrainedhasattrrr from_fileospathjoinr CONFIGURATIONtaskrrrrr updaterr KeyError) clsrkwargsrcfg task_name model_cfg export_cfgexporteres r from_modelzExporter.from_models| eS ! ! 1)%00Euk*****O  GLLI$; < <>>H 5+ & & (II 9l + + 2GIv4N4N 2&1IN 899 3 ! ! *   cj ) ) )# 7 7%j)VDDHH 7 7 7/IN//I///0056 7 7sD(( E2EE output_dirc dS)a'Export the model as onnx format files. In some cases, several files may be generated, So please return a dict which contains the generated name with the file path. Args: opset: The version of the ONNX operator set to use. output_dir: The output dir. kwargs: In this default implementation, kwargs will be carried to generate_dummy_inputs as extra arguments (like input shape). Returns: A dict contains the model name with the model file path. N)rr3opsetr*s r export_onnxzExporter.export_onnx;s rr)r2) __name__ __module__ __qualname____doc__r classmethodrrrr1rr7r5rrrrs"uUCZ0"""["H  c   ^   rr)r"abcrrtypingrrmodelscope.modelsrmodelscope.utils.configrr modelscope.utils.constantr modelscope.utils.loggerr builderr loggerrr5rrrEs ##############66666666//////......###### < < < < < s< < < < < r