
    i                     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)edsr_r16f64   )EagerModelBasec                   V    \ rS rSrS rS\R                  R                  4S jrS r	Sr
g)	EdsrModel   c                     g )N )selfs    d/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/examples/models/edsr/model.py__init__EdsrModel.__init__   s        returnc                 v    [         R                  " S5        [        SS5      n[         R                  " S5        U$ )NzLoading edsr modelr   TzLoaded edsr model)logginginfor   )r   ms     r   get_eager_modelEdsrModel.get_eager_model   s-    )*4 ()r   c                 4    Sn[         R                  " U5      4$ )N)         r   )torchrandn)r   tensor_sizes     r   get_example_inputsEdsrModel.get_example_inputs   s    &K(**r   r
   N)__name__
__module____qualname____firstlineno__r   r   nnModuler   r   __static_attributes__r
   r   r   r   r      s      +r   r   )r   r   torchsr.modelsr   
model_baser   r   r
   r   r   <module>r)      s      & '+ +r   