
    +h                         S SK Jr  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)    )	dataclass)OptionalN   )ConfigMixinregister_to_config)
ModelMixin)
BaseOutputc                   >    \ rS rSr% Sr\\R                     \S'   Sr	g)ReduxImageEncoderOutput   Nimage_embeds )
__name__
__module____qualname____firstlineno__r   r   torchTensor__annotations____static_attributes__r       `/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/pipelines/flux/modeling_flux.pyr   r      s    +/L(5<<(/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   typingr   r   torch.nnr"   configuration_utilsr   r   models.modeling_utilsr   utilsr	   r   r   r   r   r   <module>r8      sG     "    B /  0j 0 0A
K Ar   