o -wi @sddlZddlmZmZmZmZmZmZddlZddl m Z er(e dZ e dZ defddZ d ed eed eeefdeeeffd d ZdeddfddZ ddeddeeddfddZd edddefddZdS)N) TYPE_CHECKINGAnyDictListOptionalSequence) get_modulez numpy.arrayztorch.FloatTensorreturncs*tdfddtdtDS)Ncsg|] }||qSr ).0i chunk_size input_listr h/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/integration/diffusers/resolvers/utils.py szchunkify..r)maxrangelen)rrr rrchunkify s  rpipelineargskwargscCstt|jj}t|D] \}}||||d<q|D]}|d|vr/|dj||d<qd|vrQ|d}|durM||j | dnd|d<d|vrg|ddurgtdt|di|S)Nrr generator)seeddevice random_stateip_adapter_imagezIP-Adapter-Image)listinspect signature__call__ parametersitems enumeratedefault initial_seedr get_statecpunumpytolistwandblogImage)rrrpipeline_call_parametersidxargpipeline_parameterrr r rget_updated_kwargss,  r3imagenp_arraycs(tdddjfdd|DddS)Nr*PPlease ensure NumPy is installed. You can run `pip install numpy` to install it.requiredcs&g|]}j|dddqS)uint8)rr axes) transposearrayastyper imgnpr rr5s&z*postprocess_pils_to_np..raxis)rstack)r4r rBrpostprocess_pils_to_np/srGFimages normalizecCs:tddd}|rdd|Dn|}|j|j|dddd S) Nr*r6r7cSsg|] }|ddqS)r9)r?r@r r rrAsz3postprocess_np_arrays_for_video..rrD)rr r:r;)rr=rF)rHrIrCr r rpostprocess_np_arrays_for_video:s rLlatentstorch_float_tensorcCstddd}|Z|jj|jko|jjj}|r+||t t |jj j}|jj ||jjjddd}|rD|jj|jd|jdurO|j|}|jj|d d }||WdS1sgwYdS) aDecode latents generated by [`diffusers.StableDiffusionXLPipeline`](https://huggingface.co/docs/diffusers/main/en/api/pipelines/stable_diffusion/stable_diffusion_xl#stable-diffusion-xl). Args: pipeline: (diffusers.DiffusionPipeline) The Diffusion Pipeline from [`diffusers`](https://huggingface.co/docs/diffusers). latents (torch.FloatTensor): The generated latents. Returns: List of `PIL` images corresponding to the generated latents. torchzPlease ensure PyTorch is installed. You can check out https://pytorch.org/get-started/locally/#start-locally for installation instructions.r7F) return_dictr)dtypeNpil) output_type)rno_gradvaerQfloat16config force_upcast upcast_vaetonextiterpost_quant_convr#decodescaling_factor watermarkapply_watermarkimage_processor postprocessmaybe_free_model_hooks)rrMrOneeds_upcastingrHr r rdecode_sdxl_t2i_latentsEs2    $rf)F)r typingrrrrrrr, wandb.utilrr5rNrstrr3rGboolrLrfr r r rs6