ktjH^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|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 a/var/www/html/CA-Chatbot/venv/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  ,e , ,,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)rrrrr rEXTERNAL)r rrrs rhas_external_datar")s{5!!J""$$ ,  K##O44 9RVaVj9j9jttt  5rN) fusion_utilsronnxrronnx.external_data_helperrrr onnxruntimerrr"rrr(s %$$$$$((((((((777777 -z----8Zr