[Diffusion]: Diffusion Ring Attention support (#273)

Signed-off-by: mxuax <mxuax@connect.ust.hk>
Signed-off-by: Hongsheng Liu <liuhongsheng4@huawei.com>
Co-authored-by: Jiangyun Zhu <riverclouds.zhu@qq.com>
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>

tools/pre_commit/check_pickle_imports.py

@@ -19,7 +19,7 @@ ALLOWED_FILES = {
     "vllm_omni/entrypoints/omni_llm.py",
     "tests/e2e/offline_inference/utils.py",
     "vllm_omni/diffusion/distributed/group_coordinator.py",
-    "tests/diffusion/attention/test_ulysses_sequence_parallel.py",
+    "tests/diffusion/attention/test_sequence_parallel.py",
 }
 
 PICKLE_RE = re.compile(