o W+ 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)selfrre/home/app/PaddleOCR-VL-test/.venv_paddleocr/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) isinstancestrrZfrom_pretrainedhasattrrr from_fileospathjoinr Z CONFIGURATIONtaskrrrrrupdaterr KeyError) clsrkwargsrcfgZ task_nameZ model_cfgZ export_cfgZexportererrr 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)Zopsetr$rrr export_onnx;szExporter.export_onnxr)r() __name__ __module__ __qualname____doc__r classmethodrrrr'rr*rrrrr s $r )rabcrrtypingrrZmodelscope.modelsrZmodelscope.utils.configrrZmodelscope.utils.constantr Zmodelscope.utils.loggerr Zbuilderr loggerr rrrrs