o xi@sddlmZddlZddlmZddlmZmZddlZddl m Z er,e dZ e dZ d#d d Z d$ddZd%ddZ d&d'ddZd(d!d"ZdS))) annotationsN)Sequence) TYPE_CHECKINGAny) get_modulez numpy.arrayztorch.FloatTensorreturnlistcs*tdfddtdtDS)Ncsg|] }||qSr ).0i chunk_size input_listr _/home/ubuntu/.local/lib/python3.10/site-packages/wandb/integration/diffusers/resolvers/utils.py szchunkify..r)maxrangelen)rrr r rchunkifys  rpipelinerargs Sequence[Any]kwargsdict[str, Any]cCstt|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)rinspect 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 rr7s&z*postprocess_pils_to_np..raxis)rstack)r4r rBrpostprocess_pils_to_np1srGFimageslist[np_array] normalize bool | NonecCs:tddd}|rdd|Dn|}|j|j|dddd S) Nr*r6r7cSsg|] }|ddqS)r9)r?r@r r rrCsz3postprocess_np_arrays_for_video..rrD)rr r:r;)rr=rF)rHrJrCr r rpostprocess_np_arrays_for_video<s rNlatentstorch_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_gradvaerSfloat16config force_upcast upcast_vaetonextiterpost_quant_convr#decodescaling_factor watermarkapply_watermarkimage_processor postprocessmaybe_free_model_hooks)rrOrQneeds_upcastingrHr r rdecode_sdxl_t2i_latentsGs2    $rh)rr)rrrrrrrr)r4rrr5)F)rHrIrJrKrr5)rrrOrPrr) __future__rr collections.abcrtypingrrr, wandb.utilrr5rPrr3rGrNrhr r r rs