§
    ƒØj÷  ã                   óŽ   — d dl Z d dlZd dlmZ d dlZd dlZd dlZd dl	m
Z
 dd„Zedk    re j        d         Z ee¦  «         dS dS )é    N)ÚPath)ÚMultiHeadAttentionc                 óP  — t          | ¦  «        } t          j        j                             | dt
          i¬¦  «        }|j        dd …         } |t          j                             d|z   ¦  «        ¦  «         t          j	        |j        t          j
        d¬¦  «        f}|€| }nt          |¦  «        }|dz  }t          j                             |||¬¦  «        \  }}d	„ |j        j        D ¦   «         }t!          j        ||j        |j        d
œt'          |dz  d¦  «        ¦  «         d S )Nr   )Úcustom_objectsé   )r   Úinput©Únamez
model.onnx)Úinput_signatureÚoutput_pathc                 ó   — g | ]	}|j         ‘Œ
S © r	   )Ú.0Úns     úŠ/Users/tedpeters/Documents/Projects/Claude Projects/Misc Tools/notename-sheets/.venv/lib/python3.11/site-packages/oemer/convert_to_onnx.pyú
<listcomp>zconvert.<locals>.<listcomp>   s   € Ð=Ð=Ð=˜q�A”FÐ=Ð=Ð=ó    )Úoutput_namesÚinput_shapeÚoutput_shapezmetadata.pklÚwb)r   ÚtfÚkerasÚmodelsÚ
load_modelr   r   ÚnpÚrandomÚ
TensorSpecÚuint8Útf2onnxÚconvertÚ
from_kerasÚgraphÚoutputÚpickleÚdumpr   Úopen)	Ú
input_pathr   ÚmodelÚ	inp_shapeÚspecÚoutput_modelÚmodel_protoÚ_r   s	            r   r!   r!      s0  € Ý�jÑ!Ô!€Jõ ŒHŒO×&Ò& zÐCWÕYkÐBlÐ&ÑmÔm€EØÔ! ! " "Ô%€IØ	€E�"Œ)×
Ò
˜4 	™>Ñ
*Ô
*Ñ+Ô+Ð+ÝŒM˜%Ô+­R¬X¸GÐDÑDÔDÐF€DàÐØ ˆˆå˜;Ñ'Ô'ˆØ Ñ-€LÝ”_×/Ò/°ÀtÐYeÐ/ÑfÔf�N€K�Ø=Ð= KÔ$5Ô$<Ð=Ñ=Ô=€LÝ
„Kà(Ø Ô,Ø!Ô.ð	
ð 	
õ
 	ˆ[˜>Ñ)¨4Ñ0Ô0ñô ð ð ð r   Ú__main__r   )N)Úsysr%   Úpathlibr   r    Únumpyr   Ú
tensorflowr   Úomnizart.models.t2tr   r!   Ú__name__ÚargvÚ
model_pathr   r   r   ú<module>r8      s¦   ðØ 
€
€
€
Ø €€€Ø Ð Ð Ð Ð Ð à €€€Ø Ð Ð Ð Ø Ð Ð Ð à 2Ð 2Ð 2Ð 2Ð 2Ð 2ðð ð ð ð4 ˆzÒÐØ”˜!”€JØ€GˆJÑÔÐÐÐð Ðr   