
    
3j                         S SK Jr  S SKrS SKJr  SSKJrJr  SSKJ	r	  SSK
Jr  \ " S S\5      5       r " S	 S
\	\5      rg)    )	dataclassN   )ConfigMixinregister_to_config)
ModelMixin)
BaseOutputc                   >    \ rS rSr% Sr\R                  S-  \S'   Srg)ReduxImageEncoderOutput   Nimage_embeds )	__name__
__module____qualname____firstlineno__r   torchTensor__annotations____static_attributes__r       `/home/wildlama/miniconda3/lib/python3.13/site-packages/diffusers/pipelines/flux/modeling_flux.pyr
   r
      s    (,L%,,%,r   r
   c                   p   ^  \ rS rSr\  S
S\S\SS4U 4S jjj5       rS\R                  S\	4S jr
S	rU =r$ )ReduxImageEncoder   	redux_dimtxt_in_featuresreturnNc                    > [         TU ]  5         [        R                  " XS-  5      U l        [        R                  " US-  U5      U l        g )Nr   )super__init__nnLinearredux_up
redux_down)selfr   r   	__class__s      r   r    ReduxImageEncoder.__init__    s<     			)q-@A))Oa$7Ir   xc                     U R                  [        R                  R                  U R	                  U5      5      5      n[        US9$ )N)r   )r$   r!   
functionalsilur#   r
   )r%   r(   projected_xs      r   forwardReduxImageEncoder.forward+   s3    oobmm&8&8q9I&JK&K@@r   )r$   r#   )i  i   )r   r   r   r   r   intr    r   r   r
   r-   r   __classcell__)r&   s   @r   r   r      s_     #JJ J 
	J JA A*A A Ar   r   )dataclassesr   r   torch.nnr!   configuration_utilsr   r   models.modeling_utilsr   utilsr   r
   r   r   r   r   <module>r6      sD     "   B /  -j - -A
K Ar   