
    bir                         d dl mZ ddlmZmZ ddlmZmZmZ  ej                  e
      Z G d de      Z G d d	e      Z G d
 de      Zy)    )List   )	deprecatelogging   )FluxControlNetModelFluxControlNetOutputFluxMultiControlNetModelc                        e Zd Z fdZ xZS )r	   c                 B    d}t        dd|       t        |   |i | y )NzImporting `FluxControlNetOutput` from `diffusers.models.controlnet_flux` is deprecated and this will be removed in a future version. Please use `from diffusers.models.controlnets.controlnet_flux import FluxControlNetOutput`, instead.z5diffusers.models.controlnet_flux.FluxControlNetOutput0.34r   super__init__selfargskwargsdeprecation_message	__class__s       [/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/diffusers/models/controlnet_flux.pyr   zFluxControlNetOutput.__init__   s,     JI6Sfg$)&)    __name__
__module____qualname__r   __classcell__r   s   @r   r	   r	          * *r   r	   c                   t     e Zd Zddddddddd	g d
ddfdedededededededededee   dedef fdZ xZS )r   r   @      &         i   i   F)   8   r'   N
patch_sizein_channels
num_layersnum_single_layersattention_head_dimnum_attention_headsjoint_attention_dimpooled_projection_dimguidance_embedsaxes_dims_ropenum_modeconditioning_embedding_channelsc                 Z    d}t        dd|       t        | 	  |||||||||	|
||       y )NzImporting `FluxControlNetModel` from `diffusers.models.controlnet_flux` is deprecated and this will be removed in a future version. Please use `from diffusers.models.controlnets.controlnet_flux import FluxControlNetModel`, instead.z4diffusers.models.controlnet_flux.FluxControlNetModelr   )r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r   )r   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r   r   s                 r   r   zFluxControlNetModel.__init__!   sP     HH&Ref!#!/1 3 3"7+),K 	 	
r   )r   r   r   intboolr   r   r   r   s   @r   r   r       s     !#"%#%#'%( %$0/3

 
 	

 
  
 !
 !
  #
 
 S	
 
 *-
 
r   r   c                        e Zd Z fdZ xZS )r
   c                 B    d}t        dd|       t        |   |i | y )NzImporting `FluxMultiControlNetModel` from `diffusers.models.controlnet_flux` is deprecated and this will be removed in a future version. Please use `from diffusers.models.controlnets.controlnet_flux import FluxMultiControlNetModel`, instead.z9diffusers.models.controlnet_flux.FluxMultiControlNetModelr   r   r   s       r   r   z!FluxMultiControlNetModel.__init__C   s,     RMvWjk$)&)r   r   r   s   @r   r
   r
   B   r   r   r
   N)typingr   utilsr   r   controlnets.controlnet_fluxr   r	   r
   
get_loggerr   logger r   r   <module>r?      sO      & l l 
		H	%*/ *
- 
D*7 *r   