:jH\ddlmZddlmZmZddlmZddlmZddl m Z defdZ defdZ y ) ) NumpyHelper) ModelProto TensorProto)set_external_data) OnnxModel)OrtValuemodelc~g}t|}|jD]}|jD]}|j}|j ds!t j |}tj|}|j||ft|d||_|jdt|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_modelgraphrr numpy_tensor ort_values x/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/onnxruntime/transformers/onnx_utils.pyextract_raw_data_from_modelr sM5!J""$ 3 ,, 3K##D##J/*33K@ $88F $$dI%67!+ B#'  &&z2 3 3  ,e ,,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)rrrrr rEXTERNAL)r rrrs rhas_external_datar")se5!J""$ ,, K##O49R9RVaVjVj9j  rN) fusion_utilsronnxrronnx.external_data_helperrrr onnxruntimerrr"rrr(s. %(7 -z-8Zr