o
    áÙ·i¥  ã                   @   sH   d Z ddlmZmZmZmZ ddlmZmZm	Z	m
Z
mZmZ g d¢ZdS )z.Hook mechanism for model forward interception.é    )Ú	BaseStateÚHookRegistryÚ	ModelHookÚStateManager)ÚSequenceParallelGatherHookÚSequenceParallelSplitHookÚapply_sequence_parallelÚ#disable_sequence_parallel_for_modelÚ"enable_sequence_parallel_for_modelÚremove_sequence_parallel)
r   r   r   r   r   r   r   r   r
   r	   N)Ú__doc__Úvllm_omni.diffusion.hooks.baser   r   r   r   Ú+vllm_omni.diffusion.hooks.sequence_parallelr   r   r   r	   r
   r   Ú__all__© r   r   úV/home/ubuntu/.local/lib/python3.10/site-packages/vllm_omni/diffusion/hooks/__init__.pyÚ<module>   s    	