
    i                     F    S SK r S SKrS SKJrJr  SSKJr   " S S\5      rg)    N)AutoTokenizerMobileBertModel   )EagerModelBasec                   V    \ rS rSrS rS\R                  R                  4S jrS r	Sr
g)MobileBertModelExample   c                     g )N )selfs    j/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/examples/models/mobilebert/model.py__init__MobileBertModelExample.__init__   s        returnc                     [         R                  " S5        [        R                  " SSS9n[         R                  " S5        U$ )Nzloading mobilebert modelgoogle/mobilebert-uncasedF)return_dictzloaded mobilebert model)logginginfor   from_pretrained)r   models     r   get_eager_model&MobileBertModelExample.get_eager_model   s8    /0//'U
 	./r   c                 D    [         R                  " S5      nU" SSS9S   4$ )Nr   zHello, my dog is cutept)return_tensors	input_ids)r   r   )r   	tokenizers     r   get_example_inputs)MobileBertModelExample.get_example_inputs   s)    !112MN	1$GTVVr   r   N)__name__
__module____qualname____firstlineno__r   torchnnModuler   r    __static_attributes__r   r   r   r   r      s!     Wr   r   )r   r&   transformersr   r   
model_baser   r   r   r   r   <module>r,      s!      7 'W^ Wr   