o
    ¶Ïi   ã                   @   s    d dl Z G dd„ de jjƒZdS )é    Nc                       sH   e Zd Z‡ fdd„Zdejdejfdd„Zdejdejfdd	„Z‡  ZS )
ÚPixelspaceConversionVAEc                    s"   t ƒ  ¡  tj t d¡¡| _d S )Ng      ð?)ÚsuperÚ__init__ÚtorchÚnnÚ	ParameterÚtensorÚpixel_space_vae)Úself©Ú	__class__© ú7/mnt/c/Users/fbmor/ComfyUI/comfy/pixel_space_convert.pyr      s   
z PixelspaceConversionVAE.__init__ÚpixelsÚreturnc                 O   ó   |S ©Nr   )r
   r   Ú_argsÚ_kwargsr   r   r   Úencode   ó   zPixelspaceConversionVAE.encodeÚsamplesc                 O   r   r   r   )r
   r   r   r   r   r   r   Údecode   r   zPixelspaceConversionVAE.decode)	Ú__name__Ú
__module__Ú__qualname__r   r   ÚTensorr   r   Ú__classcell__r   r   r   r   r      s    r   )r   r   ÚModuler   r   r   r   r   Ú<module>   s    