o 3IhH@s`ddlmZddlmZmZddlmZddlmZddl m Z defddZ defd d Z d S) ) NumpyHelper) ModelProto TensorProto)set_external_data) OnnxModel)OrtValuemodelcCsg}t|}|D]/}|jD])}|j}|dr8t|}t|}| ||ft |dd||_| dqq t |ddiS)a Extract external data from model and return the external data as a list of tuples (name, value). Note this function does not handle external data that is not loaded into the model as raw data. Args: model (ModelProto): the model proto to extract external data from. Returns: (external_names, external_values): a tuple of two lists of external data names and values. raw_datazfoo.bin)locationstrictF) rgraphs initializernameHasFieldrto_arrayrortvalue_from_numpyappendr ClearFieldzip)r external_data onnx_modelgraphr r numpy_tensor ort_valuerc/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/onnxruntime/transformers/onnx_utils.pyextract_raw_data_from_model s         rcCsDt|}|D]}|jD]}|dr|jtjkrdSq qdS)z Check if the model has external data. Args: model (ModelProto): the model proto to check for external data. Returns: bool: True if the model has external data, False otherwise. data_locationTF)rr r rrrEXTERNAL)rrrr rrrhas_external_data)s  rN) fusion_utilsronnxrronnx.external_data_helperrrr onnxruntimerrrrrrrs