
    9i                         S SK r S SKJrJrJrJr  S SKrS SKJr  S SK	J
r
  S SKJr  S SKJrJr  S SKJr  \" 5       r\R&                  " \R(                  \R(                  S9 " S	 S
\
5      5       rg)    N)AnyDictOptionalUnion)Models)
TorchModel)MODELS)	ModelFileTasks)
get_logger)module_namec                   6   ^  \ rS rSrS\4U 4S jjrS rSrU =r$ )FFTInpainting   	model_dirc                 x  > [         T	U ]  " U40 UD6  SSKJn  UR	                  SS5      nUR	                  SS5      nU" XS9nU(       ak  [
        R                  R                  U[        R                  5      n[        R                  SU 35        [        R                  " US	S
9nUR                  USS9  X`l        g )N   )DefaultInpaintingTrainingModule
pretrainedTpredict_onlyF)r   r   zloading pretrained model from cpu)map_location)strict)super__init__defaultr   getospathjoinr
   TORCH_MODEL_FILELOGGERinfotorchloadload_state_dictmodel)
selfr   kwargsr   r   r   netr   state	__class__s
            k/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/modelscope/models/cv/image_inpainting/model.pyr   FFTInpainting.__init__   s    -f-<ZZd3
zz.%8-<77<<	9+E+EFDKK8?@JJt%8Ee4
    c                 $    U R                  U5      $ )Nr'   )r(   inputss     r-   forwardFFTInpainting.forward#   s    zz&!!r/   r1   )	__name__
__module____qualname____firstlineno__strr   r3   __static_attributes____classcell__)r,   s   @r-   r   r      s    # " "r/   r   )r   typingr   r   r   r   r$   modelscope.metainfor   'modelscope.models.base.base_torch_modelr   modelscope.models.builderr	   modelscope.utils.constantr
   r   modelscope.utils.loggerr   r"   register_moduleimage_inpaintingr    r/   r-   <module>rE      s`    	 - -  & > , 6 .	 	(?(?A"J "A"r/   