g:\ddlmZddlmZmZddlmZddlmZddl m Z defdZ defdZ y ) ) NumpyHelper) ModelProto TensorProto)set_external_data) OnnxModel)OrtValuemodelcxg}t|}|jD]}|jD]}|j}|j ds!t j |}tj|}|j||ft|d||_|jdt|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 X/var/www/openai/venv/lib/python3.12/site-packages/onnxruntime/transformers/onnx_utils.pyextract_raw_data_from_modelr sM5!J""$ ,,K##D##J/*33K@ $88F $$dI%67!+ B#'  &&z2-%  ct|}|jD]E}|jD]4}|jds|jt j k(s3yGy)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##O49R9RVaVjVj9j-% rN) fusion_utilsronnxrronnx.external_data_helperrrr onnxruntimerrr!rrr's. %(7 z8Zr