U  d @s\ddlZddlZddlmZmZddgZddgZddgZe e e e e e d d d Z d d Z dS)N) DDIMSchedulerDiffusionPipelinez)stabilityai/stable-diffusion-2-inpaintingz$runwayml/stable-diffusion-inpaintingza photo of a man.za photo of a girl.z bad, uglydeformed)dict model_pathpromptnegative_promptguidance_scalenum_inference_stepc Csz|ddd}|ddd}tj|dtjd}|dt|j j |_ | |||||||dj } | d S) NimageRGB)r maskfp16)revision torch_dtypecuda)rr mask_imagernum_inference_stepsr r) convertresizerfrom_pretrainedtorchfloat16tor from_config schedulerconfig*enable_xformers_memory_efficient_attentionimages) rrrrr r r rpipeoutputr"g/home/kadir/AI/artgan/Stable-Diffusion-ControlNet-WebUI/diffusion_webui/stable_diffusion/inpaint_app.pystable_diffusion_inpaints&   r$c Csttttjdddddd}tjttddd }tjd tdd d }tjd t dd d }tj ddd.tj dddddd}tj d dd ddd}W5QRXtj dd}W5QRXttj dd}W5QRXW5QRX|jt||||||g|dW5QRXdS)NuploadsketchZ image_uploadpilZUpload)sourcetoolelem_idtypelabelrzInpaint Model Id)choicesvaluer,Prompt)linesr.r,zNegative PromptzAdvanced OptionsF)openg?g@zGuidance Scale)minimummaximumstepr.r,d2zNum Inference Step Generator)r.Outputs)r,)fninputsoutputs)grBlocksRowColumnImageDropdownstable_inpiant_model_listTextboxstable_prompt_liststable_negative_prompt_list AccordionSliderButtonGalleryclickr$)Zinpaint_image_fileZinpaint_model_idZinpaint_promptZinpaint_negative_promptZinpaint_guidance_scaleZinpaint_num_inference_stepZinpaint_predict output_imager"r"r#stable_diffusion_inpaint_app/sl     rN) gradior>r diffusersrrrDrFrGstrintr$rNr"r"r"r#s