vllm_omni.diffusion.distributed.hsdp_utils ¶ is_transformer_block_module ¶ is_transformer_block_module(name: str, module: Any) -> bool Return True for numbered modules under transformer_blocks.