diff --git a/internlm/initialize/launch.py b/internlm/initialize/launch.py index 443a53d6..946316c2 100644 --- a/internlm/initialize/launch.py +++ b/internlm/initialize/launch.py @@ -55,6 +55,10 @@ def get_default_parser(): parser.add_argument("--backend", type=str, default="nccl", help="backend for distributed communication") parser.add_argument("--seed", type=int, default=1024) parser.add_argument("--profiling", default=False, action="store_true", help="enable/disable profiling.") + parser.add_argument("--enable_ali_topology", default=False, action="store_true", help="enable ali switch topology.") + parser.add_argument( + "--disable_volc_topology", default=False, action="store_true", help="disable volc switch topology." + ) return parser