
    
3jr                     *   S SK r SSKJrJrJr  SSKJrJr  \\\4r\\4r	Sr
SrSr\" 1 \
k\k\k5      r\ R                  R                   \ R                  R"                  \ R                  R$                  \ R                  R&                  \ R                  R(                  \ R                  R*                  \ R                  R,                  \ R                  R.                  4rS\ R                  R2                  S	\S
\ R                  R2                  S-  4S jrg)    N   )AttentionModuleMixinFeedForwardLuminaFeedForward)	AttentionMochiAttention)blockstransformer_blockssingle_transformer_blockslayersvisual_transformer_blocks)temporal_transformer_blocks)r	   r
   r   modulefqnreturnc                 H    U R                  5        H  u  p#X!:X  d  M  Us  $    g )N)named_modules)r   r   submodule_name	submodules       Q/home/wildlama/miniconda3/lib/python3.13/site-packages/diffusers/hooks/_common.py_get_submodule_from_fqnr   9   s)    %+%9%9%;!  &<     )torchmodels.attentionr   r   r   models.attention_processorr   r   _ATTENTION_CLASSES_FEEDFORWARD_CLASSES&_SPATIAL_TRANSFORMER_BLOCK_IDENTIFIERS'_TEMPORAL_TRANSFORMER_BLOCK_IDENTIFIERS$_CROSS_TRANSFORMER_BLOCK_IDENTIFIERStuple"_ALL_TRANSFORMER_BLOCK_IDENTIFIERSnnConv1dConv2dConv3dConvTranspose1dConvTranspose2dConvTranspose3dLinear	Embedding_GO_LC_SUPPORTED_PYTORCH_LAYERSModulestrr    r   r   <module>r0      s
    S S B  1EF #%67 * & +K ''Q $%*	/	0 
.& " 
HHOO	HHOO	HHOO	HH	HH	HH	HHOO	HH# EHHOO # %((//TXBX r   