
    iU                         d dl mZmZ d dlmZ ddlmZ  ej                  e      Z	 G d deee      Z
 G d de
      Zy	)
   )FluxLoraLoaderMixinTextualInversionLoaderMixin)logging   )ModularPipelinec                   d    e Zd ZdZdZed        Zed        Zed        Zed        Z	ed        Z
y)	FluxModularPipelinez
    A ModularPipeline for Flux.

    > [!WARNING] > This is an experimental feature and is likely to change in the future.
    FluxAutoBlocksc                 4    | j                   | j                  z  S Ndefault_sample_sizevae_scale_factorselfs    {/home/obispo/Crisostomo_bridge/mision_env/lib/python3.12/site-packages/diffusers/modular_pipelines/flux/modular_pipeline.pydefault_heightz"FluxModularPipeline.default_height!       ''$*?*???    c                 4    | j                   | j                  z  S r   r   r   s    r   default_widthz!FluxModularPipeline.default_width%   r   r   c                      y)N    r   s    r   r   z'FluxModularPipeline.default_sample_size)   s    r   c                     d}t        | dd       /dt        | j                  j                  j                        dz
  z  }|S )N   vaer      )getattrlenr   configblock_out_channels)r   r   s     r   r   z$FluxModularPipeline.vae_scale_factor-   s@    4%1 S)K)K%Lq%PQr   c                 j    d}t        | dd       r#| j                  j                  j                  dz  }|S )N   transformer   )r   r%   r!   in_channels)r   num_channels_latentss     r   r(   z(FluxModularPipeline.num_channels_latents4   s7    !4-#'#3#3#:#:#F#F!#K ##r   N)__name__
__module____qualname____doc__default_blocks_namepropertyr   r   r   r   r(   r   r   r   r	   r	      su     +@ @ @ @       $ $r   r	   c                       e Zd ZdZdZy)FluxKontextModularPipelinez
    A ModularPipeline for Flux Kontext.

    > [!WARNING] > This is an experimental feature and is likely to change in the future.
    FluxKontextAutoBlocksN)r)   r*   r+   r,   r-   r   r   r   r0   r0   <   s     2r   r0   N)loadersr   r   utilsr   modular_pipeliner   
get_loggerr)   loggerr	   r0   r   r   r   <module>r7      sE     H  . 
		H	%!$/+>@[ !$H2!4 2r   