o Gi @sdddlmZddlmZddlmZeeZGdddeeZ Gddde Z Gd d d e Z d S) )Flux2LoraLoaderMixin)logging)ModularPipelinec@sPeZdZdZdZeddZeddZeddZed d Z ed d Z d S)Flux2ModularPipelinez A ModularPipeline for Flux2. > [!WARNING] > This is an experimental feature and is likely to change in the future. Flux2AutoBlockscC |j|jSNdefault_sample_sizevae_scale_factorselfrf/home/ubuntu/.local/lib/python3.10/site-packages/diffusers/modular_pipelines/flux2/modular_pipeline.pydefault_height! z#Flux2ModularPipeline.default_heightcCrr r r rrr default_width%rz"Flux2ModularPipeline.default_widthcCsdS)Nrr rrrr )sz(Flux2ModularPipeline.default_sample_sizecCs.d}t|dddurdt|jjjd}|S)Nvaer)getattrlenrconfigblock_out_channels)rr rrrr -sz%Flux2ModularPipeline.vae_scale_factorcCs"d}t|ddr|jjjd}|S)N transformer)rrr in_channels)rnum_channels_latentsrrrr 4s z)Flux2ModularPipeline.num_channels_latentsN) __name__ __module__ __qualname____doc__default_blocks_namepropertyrrr r r rrrrrs    rc@ eZdZdZdZeddZdS)Flux2KleinModularPipelinez A ModularPipeline for Flux2-Klein (distilled model). > [!WARNING] > This is an experimental feature and is likely to change in the future. Flux2KleinAutoBlockscCHt|jdr |jjr dSd}t|dr"|jdur"|jjo!|jjdk}|SN is_distilledFguiderrhasattrrr,r-_enablednum_conditionsrrequires_unconditional_embedsrrrr3E z7Flux2KleinModularPipeline.requires_unconditional_embedsNr!r"r#r$r%r&r3rrrrr(< r(c@r')Flux2KleinBaseModularPipelinez A ModularPipeline for Flux2-Klein (base model). > [!WARNING] > This is an experimental feature and is likely to change in the future. Flux2KleinBaseAutoBlockscCr*r+r.r2rrrr3Zr4z;Flux2KleinBaseModularPipeline.requires_unconditional_embedsNr5rrrrr7Qr6r7N) loadersrutilsrmodular_pipeliner get_loggerr!loggerrr(r7rrrrs    $