
    ^(j                     J    d dl Z  G d de j                  j                        Zy)    Nc                        e Zd Z fdZdej
                  dej
                  fdZdej
                  dej
                  fdZ xZS )PixelspaceConversionVAEc                     t         |           t        j                  j	                  t        j
                  d            | _        y )Ng      ?)super__init__torchnn	Parametertensorpixel_space_vae)self	__class__s    C/Users/danicosta/Desktop/Flux2/ComfyUI/comfy/pixel_space_convert.pyr   z PixelspaceConversionVAE.__init__   s.    $xx11%,,s2CD    pixelsreturnc                     |S N )r   r   _args_kwargss       r   encodezPixelspaceConversionVAE.encode   s    r   samplesc                     |S r   r   )r   r   r   r   s       r   decodezPixelspaceConversionVAE.decode   s    r   )	__name__
__module____qualname__r   r   Tensorr   r   __classcell__)r   s   @r   r   r      s?    EU\\  ell %,, r   r   )r   r	   Moduler   r   r   r   <module>r"      s    
	ehhoo 	r   