a
AdB1 @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZm Z m
Z
m Z m Z d dl
mZ d dlZd dlZdd Zdd ZdgZd d
Zd d Zd
d ZdS ) N)sharedimages sd_modelssd_vaesd_models_configerrors)plaintext_to_htmlc C s | d u r dS t | \}}i d|i|}d}| D ]6\}}|dtt| dtt| d d 7 }q6t|dkrd }d
| d }d||fS ) N) r r
parametersr z
r zNothing found in the image.z)r read_info_from_imageitemsr strstriplen)imagegeninfor
infokeytextmessage r 6/var/www/html/stable-diffusion-webui/modules/extras.py run_pnginfo s"
r c
C s dd }|dkr*||p&||p&||}n(|dkr<||}n|dkrN||}nd }|d u r^d S t j| \}}|d } td td| td | t|| d S )
Nc S s$ | rt | nd }|tjkr |S d S N)r $find_checkpoint_config_near_filenamer sd_default_config)xresr r r config( s z create_config..configr z.yamlzCopying config:z from:z to:)ospathsplitextprintshutilcopyfile)
Z ckpt_result
config_sourceabcr cfgfilename_Zcheckpoint_filenamer r r
create_config' s
r/ z?cond_stage_model.transformer.text_model.embeddings.position_idsc C s |r| j tjkr| S | S r )dtypetorchfloathalf)tensorenabler r r to_halfD s r6 c C sH i }| ||fD ](}t j|d }|d u r*q||j qtj|dddS )N F)indent ensure_ascii)r checkpoints_listgetupdatemetadatajsondumps)primary_model_namesecondary_model_nametertiary_model_namer= checkpoint_namecheckpoint_infor r r
read_metadataK s rE c 1 s t jjdd dd }dd }dd }d d
}fd d }fd
d}fdd}|d |f|||f|d d fd}|| \}}}|rdnd|rdnd t j_|s|dS tj| |r|s|dS |rtj| nd |r|s|d| dS |rtj| nd d}d} |rBdt j_tdj d tj jd d } nd } |r d t j_tdj d tj jd d } d t j_t
| t j_ t
| D ]t} | tv rqd | v r| | v r| | t| | } || | | | | < nt| | | | < t j jd7 _q~ t j dj dt j_tdj d tj jd d }!td! d"t j_t
|! t j_ t
|! D ]} | rrd | v rr| | v rr| tv rq|!| }"| | }#|"j|#jkrP|"jdd |"jd#d |#jdd |#jd#d krP|"jd d$kr.|#jd d%kr.td&|"jd d$krV|#jd d'krVtd(|"jd d'kr|#jd d$kr||"d d dd$d d d d f |#|!| d d dd$d d d d f< d)} n|"jd d%kr|#jd d$ksJ d*| d+|"j d,|#j ||"d d dd$d d d d f |#|!| d d dd$d d d d f< d)}n||"|#|!| < t|!| ||!| < t j jd7 _q~ tj|
d }$|$d urtd-|$ d.t j_tj|$d d }%|% D ]*} d/| }&|&|!v rt|%| ||!|&< q~%|r(|s(|! D ]} t|!| ||!| < q| rbt| }'t |!D ] } t |'| r@|! | d q@t j j pptj!}(|d0kr| n| |rd1nd07 | rd2nd07 d3| 7 t"j#$|( })t j d4t j_td5|) d i }*| r,|r,r|*%j& r|*%j& r,|*%j& | rz|*%t'(| W n2 t)yx }+ zt*+|+d6 W Y d }+~+n
d }+~+0 0 d7|*d8< | r|
rd9j,rj,nd rj,nd |||| |
| || d:
},i fd;d<}-|- r|- r|- t'-|,|*d=< t'-|*d>< t"j#.|)\}.}/|// d?kr\t0jj1|!|)t
|*dkrR|*nd d@ n t2|!|) t3 t4 fdAdBtj5 D d }0|0r|06 t7|)| tdC|) d3 dDt j_t j8 g dEdF t9d$D dC|) S )GNz model-merge)jobc S s, | t j_t j g dd tdD | S )Nc S s g | ] }t qS r )grr< .0r. r r r
^ z1run_modelmerger..fail..r7 )r statetextinfoendrange)r r r r fail[ s
z run_modelmerger..failc S s d| | || S )Nr r )theta0theta1alphar r r weighted_sum` s z%run_modelmerger..weighted_sumc S s | | S r r )rR theta2r r r get_differencec s z'run_modelmerger..get_differencec S s | || S r r )rQ Z
theta1_2_diffrS r r r add_differencef s z'run_modelmerger..add_differencec s@ j } j }td d}t d}| d| d| d| dS )Nr r! (z) + )
model_nameround)r) r* ZMaMb)
multiplierprimary_model_infosecondary_model_infor r filename_weighted_sumi s
z.run_modelmerger..filename_weighted_sumc s8 j } j }j }t d}| d| d| d| dS )Nr! z + rX z - rY rZ )r) r* r+ M)r^ r_ r` tertiary_model_infor r filename_add_differenceq s
z0run_modelmerger..filename_add_differencec s j S r )r[ r )r_ r r filename_nothingy s z)run_modelmerger..filename_nothing)z Weighted sumzAdd differencezNo interpolationr r z)Failed: Merging requires a primary model.z+Failed: Merging requires a secondary model.z Failed: Interpolation method (z ) requires a tertiary model.Fz Loading BzLoading z...cpu) map_locationz Loading CzMerging B and Cmodelz
Merging...zMerging A and Br! r7 zPWhen merging inpainting model with a normal one, A must be the inpainting model. z\When merging instruct-pix2pix model with a normal one, A must be the instruct-pix2pix model.Tz Bad dimensions for merged layer z: A=z, B=zBaking in VAE from z
Baking in VAEzfirst_stage_model.r z .inpaintingz.instruct-pix2pix.ZSavingz
Saving to zreadin metadata from jsonptformatwebui)
typeZprimary_model_hashZsecondary_model_hashZtertiary_model_hash
interp_methodr^ save_as_half custom_namer( bake_in_vaediscard_weightsZ
is_inpaintingZis_instruct_pix2pixc s@ | | j| j| jdd d | j< | jdi d S )Nsd_merge_recipe)nameZ legacy_hashru sd_merge_models)calculate_shorthashrv hashr= r; sha256r< )rD )rw r r add_model_metadata& s z+run_modelmerger..add_model_metadataru rw z .safetensors)r= c 3 s | ]}|j kr|V qd S r )rv )rI ckpt)r- r r @ rK z"run_modelmerger..zCheckpoint saved to zCheckpoint savedc S s g | ]}t jjt d qS ))choices)rG Dropdownr< r checkpoint_tilesrH r r r rJ J rK z#run_modelmerger..):r rL begin job_countr r: rM r% r- read_state_dictr keyssampling_stepstqdm checkpoint_dict_skip_on_merger; r1
zeros_like
sampling_stepnextjobshape RuntimeErrorr6 r vae_dict
load_vae_dictrecompilelistsearchpopcmd_optsckpt_dir
model_pathr" r# joinr< r= r> loads Exceptionr displayrz r? r$ lower safetensors save_filesave list_modelsnextvaluesrx r/ rN rO )1id_taskr@ rA rB rp r^ rq rr checkpoint_formatr( rs rt
save_metadataadd_merge_recipecopy_metadata_fields
metadata_jsonrP rT rV rW ra rd re Z theta_funcsZfilename_generatorZ theta_func1Z theta_func2Zresult_is_inpainting_modelZ result_is_instruct_pix2pix_modelZtheta_1Ztheta_2r t2Ztheta_0r) r* Zbake_in_vae_filenamer Z theta_0_keyregexr Zoutput_modelnamer= eZ merge_reciper{ r. extensionZ
created_modelr )r- r^ r_ rw r` rc r run_modelmergerX s0
H H>H
"
&
r )r" r r& r> r1 r modulesr r r r r r modules.ui_commonr gradiorG safetensors.torchr r r/ r r6 rE r r r r r s