fix: Correctly display sglang_tensor_parallel_size in startup logs (#459)

Co-authored-by: gaochang <gaochang@U-19PX2WQ1-0350.local>

Author: Gao016

slime/utils/arguments.py

@@ -977,6 +977,13 @@ def add_ci_arguments(parser):
             )
             return parser
 
+        def add_sglang_tp_size():
+            temp_parser = argparse.ArgumentParser(add_help=False)
+            temp_parser.add_argument("--rollout-num-gpus-per-engine", type=int, default=1)
+            temp_args, _ = temp_parser.parse_known_args()
+            sglang_tp_size = temp_args.rollout_num_gpus_per_engine
+            return sglang_tp_size
+
         # Add custom arguments in front to prevent overwritten some slime arguments.
         if add_custom_arguments is not None:
             parser = add_custom_arguments(parser)
@@ -997,6 +1004,7 @@ def add_ci_arguments(parser):
         parser = add_reward_model_arguments(parser)
         parser = add_rollout_buffer_arguments(parser)
         parser = add_ci_arguments(parser)
+        parser.set_defaults(sglang_tensor_parallel_size=add_sglang_tp_size())
 
         # For megatron
         parser = add_custom_megatron_plugins_arguments(parser)