o
    i                     @   s0   d dl Z d dlZejjZG dd de jjZdS )    Nc                       sD   e Zd Z				ddededdf fddZdejfd	d
Z  ZS )ReduxImageEncoder     N	redux_dimtxt_in_featuresreturnc                    sL   t    || _|| _|| _tj||d |d| _tj|d ||d| _d S )N   )dtype)	super__init__r   devicer	   opsLinearredux_up
redux_down)selfr   r   r   r	   	__class__ 2/mnt/c/Users/fbmor/ComfyUI/comfy/ldm/flux/redux.pyr      s   
zReduxImageEncoder.__init__c                 C   s   |  tjj| |}|S )N)r   torchnn
functionalsilur   )r   sigclip_embedsprojected_xr   r   r   forward   s   zReduxImageEncoder.forward)r   r   NN)	__name__
__module____qualname__intr   r   Tensorr   __classcell__r   r   r   r   r      s    r   )r   	comfy.opscomfyr   manual_castr   Moduler   r   r   r   r   <module>   s    