biE j d dl mZmZ d dlmZ ddlmZ ej e Z G d deee Z
y) )FluxLoraLoaderMixinTextualInversionLoaderMixin)logging )ModularPipelinec ` e Zd ZdZed Zed Zed Zed Zed Z y)FluxModularPipelinez
A ModularPipeline for Flux.
This is an experimental feature and is likely to change in the future.
c 4 | j | j z S Ndefault_sample_sizevae_scale_factorselfs l/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/diffusers/modular_pipelines/flux/modular_pipeline.pydefault_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 | dd /dt | j j j dz
z }|S )N vaer )getattrlenr configblock_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 | dd r#| j j j dz }|S )N transformer )r r$ r in_channels)r num_channels_latentss r r' z(FluxModularPipeline.num_channels_latents6 s7 ! 4 -#'#3#3#:#:#F#F!#K ##r N)
__name__
__module__ __qualname____doc__propertyr r r
r r' r r r r r so @ @ @ @ $ $r r N) loadersr r utilsr modular_pipeliner
get_loggerr( loggerr r r r r2 s7 H .
H %#$/+>@[ #$r