[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:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user