[Diffusion] Tiny rename parallel_groups (#16743)

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
Xiaoyu Zhang
2026-01-09 15:08:25 +08:00
committed by GitHub
parent 41609b52fe
commit 9d4d57dbfa
2 changed files with 2 additions and 2 deletions

View File

@@ -405,8 +405,8 @@ def initialize_model_parallel(
assert _SP is None, "sequence parallel group is already initialized"
try:
from .yunchang import PROCESS_GROUP as _YC_PROCESS_GROUP
from .yunchang import (
from .parallel_groups import PROCESS_GROUP as _YC_PROCESS_GROUP
from .parallel_groups import (
set_seq_parallel_pg_by_sp_groups as _set_seq_parallel_pg_by_sp_groups,
)
except ImportError: