[BugFix]Fix ep size (#3092)

* fix ep

* fix num_layer
This commit is contained in:
YuanRisheng
2025-07-30 21:03:12 +08:00
committed by GitHub
parent d17886de19
commit 7dfdd157ac
4 changed files with 10 additions and 1 deletions

View File

@@ -513,6 +513,12 @@ def parse_args():
default=1,
help="expert parallel size",
)
parser.add_argument(
"--data_parallel_size",
type=int,
default=1,
help="data parallel size",
)
parser.add_argument(
"--enable_expert_parallel",
action="store_true",