[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:
co-authored by
gemini-code-assist[bot]
parent
41609b52fe
commit
9d4d57dbfa
@@ -405,8 +405,8 @@ def initialize_model_parallel(
|
|||||||
assert _SP is None, "sequence parallel group is already initialized"
|
assert _SP is None, "sequence parallel group is already initialized"
|
||||||
|
|
||||||
try:
|
try:
|
||||||
from .yunchang import PROCESS_GROUP as _YC_PROCESS_GROUP
|
from .parallel_groups import PROCESS_GROUP as _YC_PROCESS_GROUP
|
||||||
from .yunchang import (
|
from .parallel_groups import (
|
||||||
set_seq_parallel_pg_by_sp_groups as _set_seq_parallel_pg_by_sp_groups,
|
set_seq_parallel_pg_by_sp_groups as _set_seq_parallel_pg_by_sp_groups,
|
||||||
)
|
)
|
||||||
except ImportError:
|
except ImportError:
|
||||||
|
|||||||
Reference in New Issue
Block a user