o Wli@sVdZddlZddlZddlmZddlZeeZde fddZ edkr)e dSdS)z~Patches a voice ONNX model with phoneme width alignment output. Requires the onnx package to be installed (not onnxruntime). N)SetreturncsJt}|jddd|jddd|jddd|}tjtjd|js+|j|_t |j}|j r8|j n9t }|j jD]}|jd krGq?||jq?|sWtd d St|d kretd |d Stt|td tfdd|j jDrtdd St j}|_|j j|t ||jtd|jdS)zMain entry point.modelzPath to ONNX voice model)helpz--outputz/Path to write output model (default: overwrite)z --tensor-namez6Name of tensor to mark as output (default: autodetect))levelZCeilz5No ceil tensors detected. Use --tensor-name manually.z>Multiple ceil tensors detected. Use --tensor-name manually: %szDetected tensor name: %sc3s|]}|jkVqdS)N)name).0outputZceil_tensor_nameZ/home/kim/smarthome/.venv/lib/python3.10/site-packages/piper/patch_voice_with_alignment.py :szmain..z0Tensor is already marked as output. Aborting: %szSuccessfully wrote %sr)argparseArgumentParser add_argument parse_argslogging basicConfigINFOr ronnxloadZ tensor_namesetgraphnodeZop_typeupdate_LOGGERfatallennextiterinfoanyhelperZValueInfoProtorappendsave)parserargsrZceil_tensor_namesrZceil_value_infor r r mainsT        r(__main__) __doc__rrtypingrr getLogger__name__rintr(r r r r s  ;