ktjz XdZddlZddlZddlZejdZdZdZdZ dZ dS)z/Helper classes or functions for onnxrt adaptor.Nneural_compressorlc||z }d}dt||zz}d|t|z z}|dz}td||d|ddd d S) z0Progress bar for cases where tqdm can't be used.# dz Progress: [z] z.2f%)endN)intlenprint)totaliprogress bar_lengthbarspaces percentages m/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/onnxruntime/quantization/neural_compressor/util.pysimple_progress_barr sz5yHJ J)** *C JS) *FCJ :# :v : : : : : :CCCCCCcg}|D]?}t|ds J|d|j|kr||@t|dkr|dSdS)z/Helper function to find item by name in a list.namez' should have a 'name' attribute definedrN)hasattrrappendr )r item_listitemsitems r find_by_namer *sz EtV$$VV&V&V&VVV$ 9   LL    5zzA~~Qxtrcbddl}t|tjstjds)tdtdt||j r|j |j urP| |jS|j |jurP| |jS| S tj|S#t*$rJdt-|dwxYw|S)zConvert to numpy ndarrays.rNtorchzxPlease install torch to enable subsequent data type check and conversion, or reorganize your data format to numpy array.Fz!The input data for onnx model is z6, which is not supported to convert to numpy ndarrays.)r" isinstancenpndarray importlibutil find_specloggererrorexitTensordtypebfloat16detachcputofloat32numpychalfcfloatarray Exceptiontype)datar"s rto_numpyr:7s~LLL dBJ ' '~''00  LLA    GGG dEL ) ) zU^++{{}}((**--em<<BBDDDzU[(({{}}((**--el;;AACCC;;==$$&&,,.. . x~~%   zT zzzu   s 2F&F,) __doc__r&loggingr3r$ getLoggerr)MAXIMUM_PROTOBUFrr r:rrr@s&65  . / /DDD   r