File tree Expand file tree Collapse file tree 1 file changed +2
-8
lines changed Expand file tree Collapse file tree 1 file changed +2
-8
lines changed Original file line number Diff line number Diff line change @@ -354,18 +354,12 @@ def wait(
354354
355355def main ():
356356 ray .init (address = "auto" )
357- config , _ = parse_cli_args (sys .argv [1 :])
358- config_path = None
359- args = sys .argv [1 :]
357+ config , config_file = parse_cli_args (sys .argv [1 :])
360358 config .scaling = to_structured_cfg (config .scaling , ScalingConfig )
361359 # Check whether enable scaling or not
362360 if config .scaling .enable_scaling :
363- if "--config" in args :
364- idx = args .index ("--config" )
365- if idx + 1 < len (args ):
366- config_path = args [idx + 1 ]
367361 try :
368- launch_scale_common (config_path )
362+ launch_scale_common (str ( config_file ) )
369363 except Exception as e :
370364 logger .info (f"[RayLauncher] Warning: Failed to scaler.py: { e } " )
371365 ray_main (config , run_id = 0 )
You can’t perform that action at this time.
0 commit comments