
    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  S SKJr  S SKJrJr  S S	KJr  \" 5       rSS
 jr\R0                  " \R2                  \R2                  S9 " S S\5      5       rg)    N)AnyDictOptionalUnion)	OmegaConf)instantiate_from_config)Models)
TorchModel)MODELS)	ModelFileTasks)
get_loggerc                    [         R                  SU 35        [        R                  " USS9nSU;   a  [         R                  SUS    35        US   n[	        U R
                  5      nUR                  USS9u  pg[        U5      S	:  a1  U(       a*  [         R                  S
5        [         R                  U5        [        U5      S	:  a1  U(       a*  [         R                  S5        [         R                  U5        U$ )NzLoading model from cpu)map_locationglobal_stepzGlobal Step: 
state_dictF)strictr   zmissing keys:zunexpected keys:)LOGGERinfotorchloadr   modelload_state_dictlen)configckptverbosepl_sdsdr   mus           o/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/modelscope/models/cv/image_paintbyexample/model.pyload_model_from_configr$      s    
KK%dV,-JJt%0EmE-$8#9:;	|	B#FLL1E  E 2DA
1vzgO$A
1vzg&'AL    )module_namec                   6   ^  \ rS rSrS\4U 4S jjrS rSrU =r$ )StablediffusionPaintbyexample$   	model_dirc                    > [         TU ]  " U40 UD6  [        R                  " [        R
                  R                  US5      5      n[        U[        R
                  R                  US5      5      nX@l        g )Nzv1.yamlzpytorch_model.pt)	super__init__r   r   ospathjoinr$   r   )selfr*   kwargsr   r   	__class__s        r#   r-   &StablediffusionPaintbyexample.__init__(   sW    -f-Y	 BC&BGGLL,>?A
r%   c                 $    U R                  U5      $ )Nr   )r1   inputss     r#   forward%StablediffusionPaintbyexample.forward0   s    zz&!!r%   r6   )	__name__
__module____qualname____firstlineno__strr-   r8   __static_attributes____classcell__)r3   s   @r#   r(   r(   $   s    # " "r%   r(   )F)r.   typingr   r   r   r   r   	omegaconfr   paint_ldm.utilr   modelscope.metainfor	   'modelscope.models.base.base_torch_modelr
   modelscope.models.builderr   modelscope.utils.constantr   r   modelscope.utils.loggerr   r   r$   register_moduleimage_paintbyexampler(    r%   r#   <module>rL      sk    	 - -   2 & > , 6 .	$ 	F,G,GI"J "I"r%   