
    +hF                        S SK Jr  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"                  R0                  4rS	\R"                  R4                  S
\S\\R"                  R4                     4S jrg)    )OptionalN   )AttentionModuleMixinFeedForwardLuminaFeedForward)	AttentionMochiAttention)blockstransformer_blockssingle_transformer_blockslayers)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/james-whalen/.local/lib/python3.13/site-packages/diffusers/hooks/_common.py_get_submodule_from_fqnr   4   s)    %+%9%9%;!  &<     )typingr   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_GO_LC_SUPPORTED_PYTORCH_LAYERSModulestrr    r   r   <module>r0      s      S S B  1EF #%67 )p &*J ''Q $%*	/	0 
.& " 
HHOO	HHOO	HHOO	HH	HH	HH	HHOO
# EHHOO # (588??B[ r   