o ߥic @sddlZddlmZmZddlmZmZddlmZddl m Z m Z ddl m Z ddlmZdd lmZeZGd d d eZdS) N)ABCabstractmethod)DictUnion)Model)Config ConfigDict) ModelFile) get_logger)build_exporterc@sHeZdZdZd ddZedeeeffddZ e d d efd d Z dS)ExporterzNExporter base class to output model to onnx, torch_script, graphdef, etc. NcCs ||_dSN)model)selfrrM/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/exporters/base.py__init__s zExporter.__init__rc Kst|tr t|}t|dsJ|j}ttj |t j }|j }t|dr*|j}|j}t|dr;t|ds;|j|_td|ji}t|drM||j||d<z t|||}W|Stys} z td|jd|d | d } ~ ww) 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 CONFIGURATIONtaskrrrrrupdaterr KeyError) clsrkwargsrcfg task_name model_cfg export_cfgexportererrr from_models8    zExporter.from_model output_dircKsdS)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. Nr)rr/opsetr&rrr export_onnx;szExporter.export_onnxr)r.) __name__ __module__ __qualname____doc__r classmethodrrrr-rr1rrrrr s $r )rabcrrtypingrrmodelscope.modelsrmodelscope.utils.configrrmodelscope.utils.constantr modelscope.utils.loggerr builderr loggerr rrrrs