a Ad#@sddlZddlZddlZddlZddlmZddlmZiZiZ ddZ ddZ dd Z d d Z Gd d d ZGdddZddZddZddZedZddZddZddZdS)N) defaultdict)errorscCsttdSN)extra_network_registryclearextra_network_aliasesrr>/var/www/html/stable-diffusion-webui/modules/extra_networks.py initialize sr cCs|t|j<dSr)rname) extra_networkrrr register_extra_networksr cCs |t|<dSr)r)r aliasrrr register_extra_network_aliassrcCsddlm}t|dS)NrExtraNetworkHypernet)Zmodules.extra_networks_hypernetrr rrrr register_default_extra_networkss rc@seZdZdddZddZdS)ExtraNetworkParamsNcCsn|pg|_g|_i|_|jD]L}t|tr6|ddn|g}t|dkr\|d|j|d<q|j|qdS)N=r)items positionalnamed isinstancestrsplitlenappend)selfritempartsrrr __init__ s   zExtraNetworkParams.__init__cCs |j|jkSrr)rotherrrr __eq__,szExtraNetworkParams.__eq__)N)__name__ __module__ __qualname__r"r%rrrr rs rc@s$eZdZddZddZddZdS) ExtraNetworkcCs ||_dSr)r )rr rrr r"1szExtraNetwork.__init__cCstdS)aB Called by processing on every run. Whatever the extra network is meant to do should be activated here. Passes arguments related to this extra network in params_list. User passes arguments by specifying this in his prompt: Where name matches the name of this ExtraNetwork object, and arg1:arg2:arg3 are any natural number of text arguments separated by colon. Even if the user does not mention this ExtraNetwork in his prompt, the call will stil be made, with empty params_list - in this case, all effects of this extra networks should be disabled. Can be called multiple times before deactivate() - each new call should override the previous call completely. For example, if this ExtraNetwork's name is 'hypernet' and user's prompt is: > "1girl, " params_list will be: [ ExtraNetworkParams(items=["agm", "1.1"]), ExtraNetworkParams(items=["ray"]) ] NNotImplementedError)rp params_listrrr activate4szExtraNetwork.activatecCstdS)z` Called at the end of processing for housekeeping. No need to do anything here. Nr*)rr,rrr deactivateRszExtraNetwork.deactivateN)r&r'r(r"r.r/rrrr r)0sr)cCsvi}t|D]`\}}t|d}t|d}|durD|durD|}|dur^td|q||g|q|S)aDreturns a dict mapping ExtraNetwork objects to lists of arguments for those extra networks. Example input: { 'lora': [], 'lyco': [], 'hypernet': [] } Example output: { : [, ], : [] } Nz Skipping unknown extra network: ) listrrgetrlogginginfo setdefaultextend)extra_network_dataresextra_network_nameextra_network_argsr rrrr lookup_extra_networksZs  r:c Csg}t|D]b\}}z|||||Wqtyp}z&t|d|jd|WYd}~qd}~00qtD]Z\}}||vrq|z||gWq|ty}zt|d|WYd}~q|d}~00q||j dur|j j ||j |j |j |j|ddS)zcall activate for extra networks in extra_network_data in specified order, then call activate for all remaining registered networks with an empty argument listzactivating extra network z with arguments N) batch_numberpromptsseedssubseedsr6)r:rr.r Exceptionrdisplayr rscriptsafter_extra_networks_activate iterationr<r=r>)r,r6Z activatedr r9er8rrr r.~s  2* r.c Cst|}|D]L}z||Wq tyV}z t|d|jWYd}~q d}~00q tD]X\}}||vrtqbz||Wqbty}zt|d|WYd}~qbd}~00qbdS)zcall deactivate for extra networks in extra_network_data in specified order, then call deactivate for all remaining registered networkszdeactivating extra network Nz'deactivating unmentioned extra network )r:r/r?rr@r rr)r,r6datar rDr8rrr r/s,r/z<(\w+):([^>]+)>cs*ttfdd}tt||}|fS)Ncs2|d}|d}|t|dddS)Nrr:r#)grouprrr)mr argsr7rr founds  zparse_prompt..found)rr0resub re_extra_net)promptrLrrKr parse_prompts rQcCs<g}d}|D]&}t|\}}|dur(|}||q ||fSr)rQr)r<r7 extra_datarPZupdated_promptZparsed_extra_datarrr parse_promptss  rSc Cs|dur iStj|\}}|d}i}zHtj|rnt|ddd}t|}Wdn1sd0YWn6ty}zt |d|WYd}~n d}~00|S)Nz.jsonrutf8)encodingz)reading extra network user metadata from ) ospathsplitextisfileopenjsonloadr?rr@)filenamebasenameextZmetadata_filenamemetadatafilerDrrr get_user_metadatas ,(rc)r\rWrMr2 collectionsrmodulesrrrr r rrrr)r:r.r/compilerOrQrSrcrrrr s(  *$