a Ad\<@sddlZddlmZmZmZddlmZmZddlm Z ddl m Z ddl m Z mZddlmZmZmZddlmZmZgd ZGd d d eZGd d d Zede dedddedddegddedddedddeidgZedededddedddedddedddeddddedddegddedddedddeidg ZGd d!d!eZ Gd"d#d#eZ!Gd$d%d%eZ"Gd&d'd'eZ#Gd(d)d)e"Z$Gd*d+d+e#Z%Gd,d-d-eZ&Gd.d/d/e"Z'Gd0d1d1e#Z(Gd2d3d3eZ)Gd4d5d5eZ*Gd6d7d7eZ+Gd8d9d9eZ,Gd:d;d;eZ-Gdd?d?eZ/Gd@dAdAeZ0GdBdCdCeZ1iZ2ej34D]\Z5Z6ej78e5Z9e6j:rej;8ee5ee=ee6j:e6j?dDfine2>e5ee=efiqedgie2Z@iZAeBedFZCeCD]`Z5eCe5jDdGkr4eCe5ZEeZFeCe5j:durreeEjDeFeeEj:eEjGdDfiq4edhieAZHGdIdJdJeZIGdKdLdLeZJGdMdNdNeZKGdOdPdPeZLGdQdRdReZMGdSdTdTeZNGdUdVdVeZOGdWdXdXeZPGdYdZdZeZQGd[d\d\eZRGd]d^d^eZSGd_d`d`eZTGdadbdbeZUGdcddddeZVGdedfdfeZWdS)iN) BaseModelField create_model)AnyOptional)Literal) underscore) StableDiffusionProcessingTxt2Img StableDiffusionProcessingImg2Img) sd_upscalersoptsparser)DictList) selfkwargssd_modeloutpath_samples outpath_grids sampler_indexextra_generation_paramsoverlay_imagesdo_not_reload_embeddingsseed_enable_extrasprompt_for_display sampler_noise_scheduler_overrideddim_discretizec@s>eZdZUdZeed<eed<eed<eed<dZeed<dS) ModelDefz6Assistance Class for Pydantic Dynamic Model Generationfield field_alias field_type field_valueF field_excludeN) __name__ __module__ __qualname____doc__str__annotations__rr"boolr*r*:/var/www/html/stable-diffusion-webui/modules/api/models.pyrs rc@s(eZdZdZdedddZddZdS) PydanticModelGeneratorz Takes in created classes and stubs them out in a way FastAPI/Pydantic is happy about: source_data is a snapshot of the default values produced by the class params are the names of the actual keys required by __init__ N) model_namec sdddd}||_|||_fdd|jD|_|D]B}|jtt|d|d|d|d d |vrv|d nd d q>dS) NcSs|j}|dkrd}t|S)NImager') annotationr)kvr r*r*r+field_type_generator4sz=PydanticModelGenerator.__init__..field_type_generatorcSs@ttddt|}i}|D]}i|t|jj}q |S)NcSs|tuS)N)object)xr*r*r+>zMPydanticModelGenerator.__init__..merge_class_params..)listfilterinspectgetmro signature__init__ parameters)class_Z all_classesr=classesr*r*r+merge_class_params=s z;PydanticModelGenerator.__init__..merge_class_paramsc sDg|]<\}}|tvrtt||||t|jtr6dn|jdqS)N)rrr r!)API_NOT_ALLOWEDrr isinstancedefaultproperty).0r0r1r2r*r+ Gsz3PydanticModelGenerator.__init__..keytyperCexcludeF)rrr r!r") _model_nameZ _class_dataitems _model_defappendrr)rr-class_instanceZadditional_fieldsr@fieldsr*rFr+r<.s     zPydanticModelGenerator.__init__cCs6dd|jD}t|jfi|}d|j_d|j_|S)zm Creates a pydantic BaseModel from the json and overrides provided at initialization c Ss*i|]"}|j|jt|j|j|jdfqS))rCaliasrJ)rr rr!rr")rEdr*r*r+ ^sz9PydanticModelGenerator.generate_model..T)rMrrK __config__allow_population_by_field_nameallow_mutation)rrPZ DynamicModelr*r*r+generate_modelYsz%PydanticModelGenerator.generate_model)NNN)r#r$r%r&r'r<rWr*r*r*r+r,'s +r,r rEuler)rHrIrC script_name script_argsZ send_imagesTZ save_imagesFalwayson_scriptsr init_imagesdenoising_strengthg?maskZinclude_init_images)rHrIrCrJc@s8eZdZUeddddZeeed<eed<eed<dS)TextToImageResponseNr.%The generated image in base64 format.rCtitle descriptionimagesr=info r#r$r%rrdrr'r(dictr*r*r*r+r_s r_c@s8eZdZUeddddZeeed<eed<eed<dS)ImageToImageResponseNr.r`rardr=rerfr*r*r*r+rhs rhc@s|eZdZUeddddZeded<eddd dZeed <edd dd d ddZ e ed<edddd d ddZ e ed<edddd d ddZ e ed<eddd dddZ e ed<eddd dd Zeed!<edd"d d#d Zeed$<edd%d&dZeed'<ed(d)d*d+d,d-eDdZeed.<ed(d/d0d+d1d-eDdZeed2<edd3dd d d4dZe ed5<ed d6d7dZeed8<d9S):ExtrasBaseRequestrz Resize ModezzSets the resize mode: 0 to upscale by upscaling_resize amount, 1 to upscale up to upscaling_resize_h x upscaling_resize_w.ra)r resize_modeTz Show resultsz.Should the backend return the generated image?show_extras_resultszGFPGAN VisibilityrjFz@Sets the visibility of GFPGAN, values should be between 0 and 1.)rCrbgele allow_inf_nanrcgfpgan_visibilityzCodeFormer VisibilityzDSets the visibility of CodeFormer, values should be between 0 and 1.codeformer_visibilityzCodeFormer Weightz@Sets the weight of CodeFormer, values should be between 0 and 1.codeformer_weightzUpscaling Factorz?By how much to upscale the image, only used when resize_mode=0.)rCrbrmrnrcupscaling_resizeiz Target WidthzCTarget width for the upscaler to hit. Only used when resize_mode=1.)rCrbrmrcupscaling_resize_wz Target HeightzDTarget height for the upscaler to hit. Only used when resize_mode=1.upscaling_resize_hz Crop to fitz=Should the upscaler crop the image to fit in the chosen size?upscaling_cropNonez Main upscalerzEThe name of the main upscaler to use, it has to be one of this list: z , cCsg|] }|jqSr*namerEr4r*r*r+rGr6zExtrasBaseRequest. upscaler_1zSecondary upscalerzJThe name of the secondary upscaler to use, it has to be one of this list: cCsg|] }|jqSr*rzr|r*r*r+rGr6 upscaler_2zSecondary upscaler visibilityzLSets the visibility of secondary upscaler, values should be between 0 and 1.extras_upscaler_2_visibilityz Upscale firstz/Should the upscaler run before restoring faces? upscale_firstN)r#r$r%rrkrr(rlr)rpfloatrqrrrurvintrwrxjoinr r}r'r~rrr*r*r*r+ris ,,ric@s"eZdZUedddZeed<dS)ExtraBaseResponsez HTML infoz2A series of HTML tags containing the process info.rbrc html_infoN)r#r$r%rrr'r(r*r*r*r+rs rc@s$eZdZUeddddZeed<dS)ExtrasSingleImageRequestr.FImage to work on, must be a Base64 string containing the image's data.raimageNr#r$r%rrr'r(r*r*r*r+rs rc@s$eZdZUeddddZeed<dS)ExtrasSingleImageResponseNr.r`rarrr*r*r*r+rs rc@s4eZdZUedddZeed<eddZeed<dS) FileDataz File dataz!Base64 representation of the filerdataz File namerbr{N)r#r$r%rrr'r(r{r*r*r*r+rs rc@s&eZdZUedddZeeed<dS)ExtrasBatchImagesRequestImagesz1List of images to work on. Must be Base64 stringsr imageListN)r#r$r%rrrrr(r*r*r*r+rs rc@s&eZdZUedddZeeed<dS)ExtrasBatchImagesResponserz&The generated images in base64 format.rrdN)r#r$r%rrdrr'r(r*r*r*r+rs rc@s"eZdZUedddZeed<dS)PNGInfoRequestr.zThe base64 encoded PNG imagerrNrr*r*r*r+rs rc@s6eZdZUedddZeed<edddZeed<dS) PNGInfoResponsez Image infoz7A string with the parameters used to generate the imagerreZItemsz/An object containing all the info the image hadrLN) r#r$r%rrer'r(rLrgr*r*r*r+rs rc@s$eZdZUeddddZeed<dS)ProgressRequestFzSkip current imagez Skip current image serializationraskip_current_imageN)r#r$r%rrr)r(r*r*r*r+rs rc@steZdZUedddZeed<eddZeed<edd dZe ed <ed d d dZ e ed<ed dddZ e ed<d S)ProgressResponseProgressz#The progress with a range of 0 to 1rprogressz ETA in secsr eta_relativeStatezThe current state snapshotstateNz Current imagezbThe current image in base64 format. opts.show_progress_every_n_steps is required for this to work.ra current_imagez Info textzInfo text used by WebUI.textinfo) r#r$r%rrrr(rrrgrr'rr*r*r*r+rs rc@s:eZdZUeddddZeed<eddddZeed <d S) InterrogateRequestrr.rrarclipModelzThe interrogate model used.modelN)r#r$r%rrr'r(rr*r*r*r+rs rc@s$eZdZUeddddZeed<dS)InterrogateResponseNZCaptionz$The generated caption for the image.racaption)r#r$r%rrr'r(r*r*r*r+rs rc@s"eZdZUedddZeed<dS) TrainResponsez Train infoz:Response string from train embedding or hypernetwork task.rreNr#r$r%rrer'r(r*r*r*r+rs rc@s"eZdZUedddZeed<dS)CreateResponsez Create infoz;Response string from create embedding or hypernetwork task.rreNrr*r*r*r+rs rc@s"eZdZUedddZeed<dS)PreprocessResponsezPreprocess infoz(Response string from preprocessing task.rreNrr*r*r*r+rs r)rCrcOptions_option_string_actionshelpFlagsc@sPeZdZUeddZeed<eddZeeed<eddZ e eefed<dS) SamplerItemNamerr{ZAliasesaliasesroptionsN) r#r$r%rr{r'r(rrrrr*r*r*r+rs rc@sxeZdZUeddZeed<eddZeeed<eddZ eeed<eddZ eeed <ed dZ ee ed <d S) UpscalerItemrrr{ Model Namer-Path model_pathURL model_urlScalescaleN) r#r$r%rr{r'r(r-rrrrrr*r*r*r+rs rc@s eZdZUeddZeed<dS)LatentUpscalerModeItemrrr{N)r#r$r%rr{r'r(r*r*r*r+rs rc@seZdZUeddZeed<eddZeed<eddZe eed<eddZ e eed <ed dZ eed <ed dZ e eed <dS) SDModelItemTitlerrbrr-z Short hashhashz sha256 hashsha256Filenamefilenamez Config fileconfigN) r#r$r%rrbr'r(r-rrrrrr*r*r*r+rs rc@s2eZdZUeddZeed<eddZeed<dS) SDVaeItemrrr-rrN)r#r$r%rr-r'r(rr*r*r*r+rs rc@s6eZdZUeddZeed<eddZeeed<dS)HypernetworkItemrrr{rpathN) r#r$r%rr{r'r(rrr*r*r*r+rs rc@s6eZdZUeddZeed<eddZeeed<dS)FaceRestorerItemrrr{rcmd_dirN) r#r$r%rr{r'r(rrr*r*r*r+rs rc@sLeZdZUeddZeed<eddZeeed<eddZ ee ed<dS) RealesrganItemrrr{rrrrN) r#r$r%rr{r'r(rrrrr*r*r*r+r s rc@sLeZdZUeddZeed<eddZeeed<eddZ eeed<dS) PromptStyleItemrrr{PromptpromptzNegative Promptnegative_promptN) r#r$r%rr{r'r(rrrr*r*r*r+rs rc@s~eZdZUedddZeeed<edddZee ed<edd dZ ee ed <ed d dZ eed <edddZ eed<dS) EmbeddingItemStepzHThe number of steps that were used to train this embedding, if availablerstepz SD CheckpointzFThe hash of the checkpoint this embedding was trained on, if available sd_checkpointzSD Checkpoint NamezThe name of the checkpoint this embedding was trained on, if available. Note that this is the name that was used by the trainer; for a stable identifier, use `sd_checkpoint` insteadsd_checkpoint_nameShapez5The length of each individual vector in the embeddingshapeZVectorsz&The number of vectors in the embeddingvectorsN) r#r$r%rrrrr(rr'rrrr*r*r*r+rs rc@sFeZdZUedddZeeefed<edddZ eeefed<dS) EmbeddingsResponseZLoadedz'Embeddings loaded for the current modelrloadedZSkippedzUEmbeddings skipped for the current model (likely due to architecture incompatibility)skippedN) r#r$r%rrrr'rr(rr*r*r*r+rs rc@s6eZdZUedddZeed<edddZeed<dS) MemoryResponseRAMzSystem memory statsrramCUDAznVidia CUDA memory statscudaN)r#r$r%rrrgr(rr*r*r*r+r"s rc@s:eZdZUeddddZeed<eddddZeed<dS) ScriptsListNZTxt2imgzTitles of scripts (txt2img)ratxt2imgZImg2imgzTitles of scripts (img2img)img2img)r#r$r%rrr7r(rr*r*r*r+r's rc@seZdZUeddddZeed<eddddZee ed<edd d dZ ee ed <edd d dZ ee ed <edd ddZ ee ed<eddddZ eeeed<dS) ScriptArgNLabelzName of the argument in UIralabelValuezDefault value of the argumentvalueZMinimumz+Minimum allowed value for the argumentin UIminimumz+Maximum allowed value for the argumentin UImaximumz,Step for changing value of the argumentin UIrZChoicesz Possible values for the argumentchoices)r#r$r%rrr'r(rrrrrrrrr*r*r*r+r,s rc@sheZdZUeddddZeed<eddddZeed<edd d dZ eed <ed d dZ e e ed<dS) ScriptInfoNrz Script namerar{Z IsAlwaysonz9Flag specifying whether this script is an alwayson script is_alwaysonZ IsImg2imgz8Flag specifying whether this script is an img2img script is_img2img ArgumentszList of script's argumentsrargs) r#r$r%rr{r'r(rr)rrrrr*r*r*r+r5s r)r)r)Xr9pydanticrrrtypingrrtyping_extensionsrZ inflectionrZmodules.processingr r Zmodules.sharedr r r rrrArr,r'r7r)rgrWZ#StableDiffusionTxt2ImgProcessingAPIrZ#StableDiffusionImg2ImgProcessingAPIr_rhrirrrrrrrrrrrrrrrrP data_labelsrLrHmetadatargetrrCtypemaprIZoptTypeupdaterZ OptionsModelflagsvars_optionsdestflag_typerZ FlagsModelrrrrrrrrrrrrrrrr*r*r*r+s   ?                  & $ $