o Ô>õg¹ã@s"ddlZGdd„dƒZdeiZdS)éNc@s0eZdZedd„ƒZdZdZdZdZdd„Z d S) ÚInstructPixToPixConditioningcCsddddddœiS)NÚrequired)Ú CONDITIONING)ÚVAE)ÚIMAGE)ÚpositiveÚnegativeÚvaeÚpixels©)Úsr r úA/home/comdoleger1/zen-flux-style-shape/comfy_extras/nodes_ip2p.pyÚ INPUT_TYPESs ýz(InstructPixToPixConditioning.INPUT_TYPES)rrÚLATENT)rrÚlatentÚencodezconditioning/instructpix2pixcCs|jddd}|jddd}|jd|ks |jd|krF|jddd}|jddd}|dd…|||…|||…dd…f}| |¡} i} t | ¡| d<g} ||fD]#} g} | D]}|d ¡}| |d<|d|g}|  |¡q`|  | ¡qZ| d| d| fS)NéééÚsamplesÚconcat_latent_imager)ÚshaperÚtorchÚ zeros_likeÚcopyÚappend)Úselfrrr r ÚxÚyÚx_offsetÚy_offsetÚ concat_latentÚ out_latentÚoutÚ conditioningÚcÚtÚdÚnr r r rs&(      z#InstructPixToPixConditioning.encodeN) Ú__name__Ú __module__Ú __qualname__Ú classmethodrÚ RETURN_TYPESÚ RETURN_NAMESÚFUNCTIONÚCATEGORYrr r r r rs  r)rrÚNODE_CLASS_MAPPINGSr r r r Ús)ÿ