o .e @sddlmZddlZddlZddlZddlZddlZ ddl Z ddl m Z ddl mZddlmZmZddlmZddlmZmZmZddlZddl Z ddlmZdd lmZdd lmZd Ze jsged 7Ze ej!j"Z#e joye$d dkZ%e&e$ddZ'e$ddkZ(e j)Z*dZ+ej,e+ddZ-ej,e+ddZ.ej,e+ddZ/ej0e+ddZ1de1d<e2e1Z3ej,e+ddZ4ej,e+ddZ5eddddd Z6ej78d!dZ9e jr!d"Z:ee:Z;e3je?e=dkre@d%e@e=e?e>dkre@d&e@e>e e-e.e/e3e6e4e5d'ZAeAjBd(e*d)ZAe(r!e jCeAj3d*d+d,eA_3dhd2d3ZD 4 4 5 6 6didjd@dAZEdBdCgZFejGdDdEZHeIeejJdFdGe$dHdkdIeK8eLejMdJd#dKdLd#dMZNejOdNddOZPWdn 1svwYejQdPd#dQdRgdSZRWdn 1swYejSdTd#dUkejTdVde#dKddWZUejVdXd+dYZWeLejTdZde'd[d4dWZXejTd\de'd[d4dWZYWdn 1swYeLejTd]dKd^d_d5dWZZejTd`dKdadKd6dWZ[Wdn 1swYWdn 1s wYej\eFeNeReEe%dbej]eNj^ePj_geDeUeWgeUd#d#dcj`eEeNeUeXeYeZe[geRdddeWdn 1sCwYeadfkrUeHjbd+dgdSdS)k) annotationsN)LatentConsistencyModelPipeline) LCMScheduler) AutoencoderKLUNet2DConditionModel)StableDiffusionSafetyChecker) CLIPTokenizer CLIPTextModelCLIPImageProcessor)tqdm) load_file)hf_hub_downloadz# Latent Consistencyu;

Running on CPU 🥶 This demo does not work on CPU.

CACHE_EXAMPLES1MAX_IMAGE_SIZE512USE_TORCH_COMPILEzdigiplay/DreamShaper_7vae) subfolder text_encoder tokenizerunettime_cond_proj_dimsafety_checkerfeature_extractorg_QK?g~jt? scaled_linearepsilon) beta_startbeta_end beta_scheduleprediction_typeHF_TOKENz#./LCM_Dreamshaper_v7_4k.safetensorsF)strictz missing keys:zunexpected keys:)rrrr schedulerrrcuda)devicedtypezreduce-overheadT)mode fullgraphseedintrandomize_seedboolreturncCs|rtdt}|S)Nr)randomrandintMAX_SEED)r*r,r2-/home/patrick/Latent_Consistency_Model/app.pyrandomize_seed_fnKs r4 @promptstrwidthheightguidance_scalefloatnum_inference_steps num_imagesPIL.Image.Imagec Cs$t|t||||||dddjS)N2pil)r8r:r;r<r>num_images_per_promptlcm_origin_steps output_type)torch manual_seedpipeimages)r8r*r:r;r<r>r?r2r2r3generateQs  rJzEAstronaut in a jungle, cold color palette, muted colors, detailed, 8kz!An astronaut riding a green horsez style.css)csszDuplicate Space for private usezduplicate-buttonSHOW_DUPLICATE_BUTTON)valueelem_idvisiblePromptzEnter your prompt)label show_label max_lines placeholder containerRun)scalezGenerated imagesgallery)rRrSrNgridzAdvanced options)openSeed)rRminimummaximumsteprMzRandomize seed)rRrMWidth HeightzGuidance scale for baseg?z"Number of inference steps for base)examplesinputsoutputsfncache_examples)triggersrirgrhqueueapi_namerun)rirgrhrm__main__)share)r*r+r,r-r.r+)rr5r5r6r7r7)r8r9r*r+r:r+r;r+r<r=r>r+r?r+r.r@)c __future__rosr/gradiogrnumpynp PIL.ImagePILrF lcm_pipeliner lcm_schedulerr diffusersrr3diffusers.pipelines.stable_diffusion.safety_checkerr transformersrr r r safetensors.torchr huggingface_hubr DESCRIPTIONr% is_availableiinfoint32maxr1getenvrr+rrfloat32DTYPEmodel_idfrom_pretrainedrrr load_configconfig from_configrrrr$environgetr" lcm_unet_ckptckptload_state_dictmulenprintrHtocompiler4rJrfBlocksdemoMarkdownDuplicateButtonGroupRowTextr8Button run_buttonGalleryresult AccordionSliderr*Checkboxr,r:r;r<r>Examplesonsubmitclickthen__name__launchr2r2r2r3s>                     0  d