o
yfB1 @ 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 rdS t | \}}i d|i|}d}| D ]\}}|dtt| dtt| d d 7 }qt|dkrEd }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 _C:\Users\Luke\Documents\Stable diffusion\Automatic1111\stable-diffusion-webui\modules\extras.py run_pnginfo s"
r c
C s dd }|dkr||p||p||}n|dkr ||}n |dkr'||}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 rq||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}|| \}}}|rHdnd|rMdnd t j_|sX|dS tj| |re|se|dS |rltj| nd |rz|sz|d| dS |rtj| nd d}d} |rdt j_tdj d tj jd d } nd } |rd t j_tdj d tj jd d } d t j_t
| t j_ t
| D ]5} | tv rqd | v r| | v r| | t| | } || | | | | < n t| | | | < t j jd7 _q~ t j dj dt j_tdj d tj jd d }!td! d"t j_t
|! t j_ t
|! D ]} | r2d | v r2| | v r2| tv rPq9|!| }"| | }#|"j|#jkr!|"jdd |"jd#d |#jdd |#jd#d kr!|"jd d$kr|#jd d%krtd&|"jd d$kr|#jd d'krtd(|"jd d'kr|#jd d$kr||"d d dd$d d d d f |#|!| d d dd$d d d d f< d)} nN|"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 _q9~ tj|
d }$|$d urvtd-|$ d.t j_tj|$d d }%|% D ]} d/| }&|&|!v rst|%| ||!|&< q_~%|r|s|! D ] } t|!| ||!| < q| rt| }'t |!D ]} t |'| r|! | d qt j j ptj!}(|d0kr| n| |rd1nd07 | rd2nd07 d3| 7 t"j#$|( })t j d4t j_td5|) d i }*| r|rr|*%j& r|*%j& r|*%j& | r:z
|*%t'(| W n t)y5 }+ z t*+|+d6 W Y d }+~+nd }+~+ww d7|*d8< | r|
rd9j,rIj,nd rPj,nd |||| |
| || d:
},i fd;d<}-|- rp|- rw|- t'-|,|*d=< t'-|*d>< t"j#.|)\}.}/|// d?krt0jj1|!|)t
|*dkr|*nd d@ nt2|!|) 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
^ s 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 )rQ theta2r r r get_differencec s z'run_modelmerger..get_differencec S s | || S r r )rP Z
theta1_2_diffrR 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 + rW z - rX rY )r) r* r+ M)r] r^ r_ tertiary_model_infor r filename_add_differenceq s
z0run_modelmerger..filename_add_differencec s j S r )rZ 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_hashrt sd_merge_models)calculate_shorthashru hashr= r; sha256r< )rD )rv r r add_model_metadata& s z+run_modelmerger..add_model_metadatart rv z .safetensors)r= c 3 s | ]
}|j kr|V qd S r )ru )rI ckpt)r- r r @ s 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 s z#run_modelmerger..):r rK begin job_countr r: rL 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 displayry r? r$ lower safetensors save_filesave list_modelsnextvaluesrw r/ rM rN )1id_taskr@ rA rB ro r] rp rq checkpoint_formatr( rr rs
save_metadataadd_merge_recipecopy_metadata_fields
metadata_jsonrO rS rU rV r` rc rd 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_reciperz r. extensionZ
created_modelr )r- r] r^ rv r_ rb r run_modelmergerX s8
H H>H
&
r )r" r r& r> r1 r modulesr r r r r r modules.ui_commonr gradiorG Zsafetensors.torchr r r/ r r6 rE r r r r r s