
    iP                     B    S SK r S SKrS SKJr  SSKJr   " S S\5      rg)    N)models   )EagerModelBasec                   V    \ rS rSrS rS\R                  R                  4S jrS r	Sr
g)Wav2LetterModel   c                 .    SU l         SU l        SU l        g )N
   i  i   
batch_sizeinput_frames
vocab_size)selfs    j/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/examples/models/wav2letter/model.py__init__Wav2LetterModel.__init__   s        returnc                     [         R                  " S5        [        R                  " U R                  S9n[         R                  " S5        U$ )NzLoading wav2letter model)num_classeszLoaded wav2letter model)logginginfor   
Wav2Letterr   )r   
wav2letters     r   get_eager_modelWav2LetterModel.get_eager_model   s6    /0&&4??C
./r   c                 b    U R                   SU R                  4n[        R                  " U5      4$ )N   )r   r   torchrandn)r   input_shapes     r   get_example_inputs"Wav2LetterModel.get_example_inputs   s*    4+<+<=K(**r   r   N)__name__
__module____qualname____firstlineno__r   r   nnModuler   r"   __static_attributes__ r   r   r   r      s     
 +r   r   )r   r   
torchaudior   
model_baser   r   r+   r   r   <module>r.      s       '+n +r   