FXeddlmZddlZddlmZddlmZddlm Z ddl m Z ddl m Z mZddlmZdd l mZdd l mZdd lmZd aee jZdadad adadefdZdeddfdZdS))AnyN)LCMDiffusionSetting)Context) InterfaceType)Settings)LCM_DEFAULT_MODELLCM_DEFAULT_MODEL_OPENVINO)is_reshape_required) AppSettings)DEVICE)enable_openvino_controlsTreturnc >t} |rt} |dkrdnd} t| |||||||||| |  } t| } d}|r*t t |t |t| t|}t | |t}|a|a| a|a|S)NTF) lcm_model_idprompt image_height image_widthinference_stepsguidance_scalenumber_of_imagesseed use_openvinouse_safety_checkeruse_seeduse_tiny_auto_encoder)lcm_diffusion_setting) rr rrr previous_widthprevious_heightprevious_model_idprevious_num_of_imagescontextgenerate_text_to_imager )rrrrr num_imagesrrrtiny_auto_encoder_checkboxmodel_idrlcm_diffusion_settingssettingsreshapeimagess CF:\dev\push\faster\fastsdcpu\src\frontend\webui\text_to_image_ui.pyr$r$s!H.-rz,ttuH0!'%# !-8   4HG  %       "     + +F !N"O ' M app_settingsctj5tj5tj5d}tj5tjddd}tjddd }dddn #1swxYwYtjd d d d d }tjddddd}tjddddd}tjd dd d d}tjdd5tjddddd}tjdddd} tj d d!d!"} tj d#dt"} tj d$d!d!"} tj d%dd!"} ||||||| | | | g }dddn #1swxYwYdddn #1swxYwYtj5tj d&d!d'd()}dddn #1swxYwYdddn #1swxYwYdddn #1swxYwY| || *| t||+dS),Ncvt ad}tsd}tjt |S)Nr*) interactivevalue)random_enabledgrNumberupdate)seed_vals r, random_seedz)get_text_to_image_ui..random_seedYsC)7%7N!H)&#%9++(6$6h,r-z$Describe the image you'd like to seezA fantasy landscape)labellines placeholderGenerategenerate_buttonr)elem_idscalezInference Steps)r3stepr;iiz Image Heightz Image Width2zNumber of images to generatezAdvanced optionsF)openg?g@g?zGuidance ScaleSeedr)r;r3 precisionr2zUse random seedT)r;r3r2z Use OpenVINOzUse Safety CheckerzUse tiny auto encoder for SDzGenerated imagesgallery)r; show_labelr@columns)fnoutputs)rOinputsrP)r5BlocksRowColumnTextboxButtonSlider Accordionr6Checkboxr Gallerychangeclickr$)r.r9r generate_btnnum_inference_stepsrrr%rr seed_checkboxopenvino_checkboxsafety_checker_checkboxr& input_paramsoutputs r,get_text_to_image_uirdTs __ VXX^ ^ U U VXX  ZD$9F $&9" 1$$$L               ')ir2C'''# "yCc    !iCc  Y8  \"45AAA--%'YS#=M&&&N9$ "#$) D %'K/"$(%%%M )+ ,#$<$>$>)))% /1k2"$(///+ 24<#$(222.$#+&")/2 $LE---------------QU U U U U U U U U U U U U U U n  ,#%                q^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ _______________BK666 !sI H6G/B  GB GB A=GB&G 9 GG G G G H6G H6#G $H6:H H6H# #H6&H# 'H6* I 6H: :I =H: >I  II)typingrgradior5#backend.models.lcmdiffusion_settingrr#rmodels.interface_typesrr.r constantsrr frontend.utilsr r r r r4WEBUIrr r!r"r$rdr-r,rmsKCCCCCC000000!!!!!!CCCCCCCC......$$$$$$333333 '-% & &: ::::zg{gtggggggr-