
    bi+                     N   d dl mZ ddlmZmZmZmZ i Z e       rddged<   ddged	<   d
ged<   dged<   dged<   dged<   dged<   dged<   dged<   dged<   dged<   dged<   dged<   d ged!<   d"ged#<   d$ged%<   d&ged'<   d(ged)<   d*ged+<   d,ged-<   d.ged/<   d0ged1<   d2d3ged4<   d5d6ged7<   d8ged9<   d:d;ged<<   d=ged><   d?ged@<   dAdBgedC<   dDgedE<   dFgedG<   dHgedI<   dJgedK<   dLgedM<   dNgedO<   dPgedQ<   dRgedS<   dTgedU<   dVgedW<   dXgedY<   dZged[<   d\ged]<   d^ged_<   d`geda<   dbgedc<   ddgede<   dfgedg<   dhgedi<   djgedk<   dlgedm<   dngedo<   dpgedq<   drgeds<   dtgedu<   dvgedw<   dxgedy<   dzged{<   d|ged}<   d~ged<   dged<   dged<   dged<   dged<   dged<   dged<   dged<   dged<   dged<   dged<   dged<   dged<   dged<   dged<   ddged<   dged<   dged<   dged<    e       rdged<   dged<   dged<   eser e       rddlm	Z	m
Z
 ddlmZmZ ddlmZ ddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z! ddl"m#Z# ddl$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2 ddl3m4Z4 ddl5m6Z6 ddl7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZY ddlZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZambZbmcZcmdZdmeZe  e       rddl$mfZf ddlZmgZg ddlhmiZi yyd dljZj eek el       d   eem      ejj                  ek<   y)    )TYPE_CHECKING   )DIFFUSERS_SLOW_IMPORT_LazyModuleis_flax_availableis_torch_availableMultiAdapter
T2IAdapteradapterAttentionBackendNameattention_backendattention_dispatch	AutoModel
auto_modelAsymmetricAutoencoderKLz autoencoders.autoencoder_asym_klAutoencoderDCzautoencoders.autoencoder_dcAutoencoderKLzautoencoders.autoencoder_klAutoencoderKLAllegroz#autoencoders.autoencoder_kl_allegroAutoencoderKLCogVideoXz%autoencoders.autoencoder_kl_cogvideoxAutoencoderKLCosmosz"autoencoders.autoencoder_kl_cosmosAutoencoderKLHunyuanVideoz)autoencoders.autoencoder_kl_hunyuan_videoAutoencoderKLLTXVideozautoencoders.autoencoder_kl_ltxAutoencoderKLMagvitz"autoencoders.autoencoder_kl_magvitAutoencoderKLMochiz!autoencoders.autoencoder_kl_mochiAutoencoderKLQwenImagez%autoencoders.autoencoder_kl_qwenimageAutoencoderKLTemporalDecoderz,autoencoders.autoencoder_kl_temporal_decoderAutoencoderKLWanzautoencoders.autoencoder_kl_wanAutoencoderOobleckz autoencoders.autoencoder_oobleckAutoencoderTinyzautoencoders.autoencoder_tinyConsistencyDecoderVAEz$autoencoders.consistency_decoder_vaeVQModelzautoencoders.vq_model
CacheMixincache_utilsControlNetModelzcontrolnets.controlnetFluxControlNetModelFluxMultiControlNetModelzcontrolnets.controlnet_fluxHunyuanDiT2DControlNetModel HunyuanDiT2DMultiControlNetModelzcontrolnets.controlnet_hunyuanSanaControlNetModelzcontrolnets.controlnet_sanaSD3ControlNetModelSD3MultiControlNetModelzcontrolnets.controlnet_sd3SparseControlNetModelz!controlnets.controlnet_sparsectrlControlNetUnionModelzcontrolnets.controlnet_unionControlNetXSAdapterUNetControlNetXSModelzcontrolnets.controlnet_xsMultiControlNetModelzcontrolnets.multicontrolnetMultiControlNetUnionModelz!controlnets.multicontrolnet_unionImageProjection
embeddings
ModelMixinmodeling_utilsAuraFlowTransformer2DModelz$transformers.auraflow_transformer_2dCogVideoXTransformer3DModelz%transformers.cogvideox_transformer_3dConsisIDTransformer3DModelz$transformers.consisid_transformer_3dDiTTransformer2DModelztransformers.dit_transformer_2dDualTransformer2DModelz transformers.dual_transformer_2dHunyuanDiT2DModelz#transformers.hunyuan_transformer_2dLatteTransformer3DModelz!transformers.latte_transformer_3dLuminaNextDiT2DModelztransformers.lumina_nextdit2dPixArtTransformer2DModelz"transformers.pixart_transformer_2dPriorTransformerztransformers.prior_transformerSanaTransformer2DModelztransformers.sana_transformerStableAudioDiTModelz%transformers.stable_audio_transformerT5FilmDecoderz transformers.t5_film_transformerTransformer2DModelztransformers.transformer_2dAllegroTransformer3DModelz transformers.transformer_allegroChromaTransformer2DModelztransformers.transformer_chromaCogView3PlusTransformer2DModelz%transformers.transformer_cogview3plusCogView4Transformer2DModelz!transformers.transformer_cogview4CosmosTransformer3DModelztransformers.transformer_cosmosEasyAnimateTransformer3DModelz$transformers.transformer_easyanimateFluxTransformer2DModelztransformers.transformer_fluxHiDreamImageTransformer2DModelz&transformers.transformer_hidream_imageHunyuanVideoTransformer3DModelz&transformers.transformer_hunyuan_video'HunyuanVideoFramepackTransformer3DModelz0transformers.transformer_hunyuan_video_framepackLTXVideoTransformer3DModelztransformers.transformer_ltxLumina2Transformer2DModelz transformers.transformer_lumina2MochiTransformer3DModelztransformers.transformer_mochiOmniGenTransformer2DModelz transformers.transformer_omnigenQwenImageTransformer2DModelz"transformers.transformer_qwenimageSD3Transformer2DModelztransformers.transformer_sd3SkyReelsV2Transformer3DModelz$transformers.transformer_skyreels_v2TransformerTemporalModelz!transformers.transformer_temporalWanTransformer3DModelztransformers.transformer_wanWanVACETransformer3DModelz!transformers.transformer_wan_vaceUNet1DModelzunets.unet_1dUNet2DModelzunets.unet_2dUNet2DConditionModelzunets.unet_2d_conditionUNet3DConditionModelzunets.unet_3d_conditionI2VGenXLUNetzunets.unet_i2vgen_xlKandinsky3UNetzunets.unet_kandinsky3MotionAdapterUNetMotionModelzunets.unet_motion_model UNetSpatioTemporalConditionModelz$unets.unet_spatio_temporal_conditionStableCascadeUNetzunets.unet_stable_cascadeUVit2DModelzunets.uvit_2dFlaxControlNetModelzcontrolnets.controlnet_flaxFlaxUNet2DConditionModelzunets.unet_2d_condition_flaxFlaxAutoencoderKLvae_flax   )r	   r
   )r   r   )r   )r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   )r"   )r$   r-   r.   r%   r&   r'   r(   r0   r1   r)   r*   r+   r,   r/   )r2   )r4   )"rD   r6   rE   r7   rF   rG   r8   rH   r9   r:   rI   rJ   rK   r;   rM   rL   r<   rN   rO   r=   rP   rQ   r>   r?   rR   r@   rS   rT   rA   rB   rC   rU   rV   rW   )r\   r]   r^   ra   rX   rZ   rY   r[   r_   r`   rb   )rc   )rd   )re   N__file__)module_spec)otypingr   utilsr   r   r   r   _import_structurer   r	   r
   r   r   r   r   r   autoencodersr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r#   r"   controlnetsr$   r-   r.   r%   r&   r'   r(   r0   r1   r)   r*   r+   r,   r/   r3   r2   r5   r4   transformersrD   r6   rE   r7   rF   rG   r8   rH   r9   r:   rI   rJ   rK   r;   rM   rL   r<   rN   rO   r=   rP   rQ   r>   r?   rR   r@   rS   rT   rA   rB   rC   rU   rV   rW   unetsr\   r]   r^   ra   rX   rZ   rY   r[   r_   r`   rb   rc   rd   rf   re   sys__name__globals__spec__modules     T/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/diffusers/models/__init__.py<module>ry      s   !   $2L#Ai /EGZ.[*+'2ml#=V<W898G7H348G7H34@V?W;<BZA[=>?T>U:;FaEbAB<S;T78?T>U:;>R=S9:BZA[=>IgHhDE<N;O78=Q<R89:K9L56AX@Y<=2;-.(4~m$3D2E./8MOi7j34%*;67 9N7N347KMf6g23>U=V9:9O8P456KMd5e128N7O34>Y=Z9:'8&9l#+7.&'A]@^<=B_A`=>A]@^<=<S;T78=U<V89@S?T;<>W=X9::P9Q56?Y>Z:;;M:N67:R9S56BWAX=>=L<M898L7M34=X<Y89<V;W78BbAc=>>Z=[9:<V;W78A`@a<=:R9S56CcBd>?CcBd>?MvLwHI9U8V45=X<Y89;T:U67=X<Y89?\>]:;9P8Q45A_@`<=>X=Y9:9P8Q45>Y=Z9:*7o&*7o&4J3K/04J3K/01?0@,-2B1C-.4CEV3W/0Ac@d<=6I5J12*7o&8M7N349S8T45%8$9j! )5O)	
 	
 	
 	
 	
& 	,	
 	
 	
 	
  	0.#	
 #	
 #	
 #	
 #	
 #	
 #	
 #	
 #	
H	
 	
 	
 	
 43/  '')J2GIZhpqCKKrw   