Ng:^ddlmZddlmZmZddlmZddlmZddl m Z defdZ defdZ d S) ) NumpyHelper) ModelProto TensorProto)set_external_data) OnnxModel)OrtValuemodelcg}t|}|D]}|jD]}|j}|drlt j|}tj|}| ||ft|d||_| dt|S)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)location) rgraphs initializernameHasFieldrto_arrayrortvalue_from_numpyappendr ClearFieldzip)r external_data onnx_modelgraphrr numpy_tensor ort_values _/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnxruntime/transformers/onnx_utils.pyextract_raw_data_from_modelr sM5!!J""$$ 3 3 , 3 3K#D##J// 3*3K@@ $8FF $$dI%6777!+ BBBB#'  &&z222 3  ct|}|D]:}|jD]0}|dr|jt jkrdS1;dS)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 rrrrEXTERNAL)r rrrs rhas_external_datar!)s{5!!J""$$ ,  K##O44 9RVaVj9j9jttt  5rN) fusion_utilsronnxrronnx.external_data_helperrrr onnxruntimerrr!rrr's %$$$$$((((((((777777 z8Zr