From f1b5691f1be6f07c837e78aa4458df7ed3779b64 Mon Sep 17 00:00:00 2001
From: Hz_Zhang <47402297+HaozheZhang6@users.noreply.github.com>
Date: Sun, 12 Jul 2026 11:09:34 -0700
Subject: [PATCH] Re-export get_sequence_parallel_rank and get_sp_group from
 xfuser utils

---
 diffsynth/utils/xfuser/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/diffsynth/utils/xfuser/__init__.py b/diffsynth/utils/xfuser/__init__.py
index 8e5faca2f..cf6025b7e 100644
--- a/diffsynth/utils/xfuser/__init__.py
+++ b/diffsynth/utils/xfuser/__init__.py
@@ -1 +1 @@
-from .xdit_context_parallel import usp_attn_forward, usp_dit_forward, usp_vace_forward, get_sequence_parallel_world_size, initialize_usp, get_current_chunk, gather_all_chunks
+from .xdit_context_parallel import usp_attn_forward, usp_dit_forward, usp_vace_forward, get_sequence_parallel_world_size, get_sequence_parallel_rank, get_sp_group, initialize_usp, get_current_chunk, gather_all_chunks