fix ci

Author: YanhuiDua

tests/ray/test_update_weight.py

@@ -109,7 +109,7 @@ def test_lmdeploy_update_weight_and_generate(self):
         sample_params = SampleParams(temperature=0.0, max_tokens=128, top_k=1)
 
         # init rollout_update
-        rollout_controller = RolloutController.remote(
+        rollout_controller = ray.remote(RolloutController).remote(
             self.rollout_cfg,
             self.pg,
         )
@@ -129,7 +129,7 @@ def test_lmdeploy_update_weight_and_generate(self):
 
         # init rollout_ref
         self.rollout_cfg.skip_load_weights = False
-        rollout_controller_ref = RolloutController.remote(
+        rollout_controller_ref = ray.remote(RolloutController).remote(
             self.rollout_cfg,
             self.pg,
         )

xtuner/v1/ray/base/accelerator.py

@@ -404,7 +404,6 @@ def from_placement_group(cls, worker_cls, worker_config, pg: PlacementGroup):
         rank_bundle_idx_list = []
         for rank, bundle_idx in enumerate(sorted_bundle_idxs):
             worker = worker_cls.options(
-                max_concurrency=int(os.environ.get("RAY_MAX_CONCURRENCY", 1000)),
                 placement_group=pg,
                 placement_group_bundle_index=bundle_idx,
                 **pg_options,