Improve code style of sampler (#1168)

2024-08-21 16:48:24 -07:00
parent ac1b74fa85
commit 83e23c69b3
10 changed files with 268 additions and 194 deletions
--- a/python/sglang/srt/model_executor/model_runner.py
+++ b/python/sglang/srt/model_executor/model_runner.py
@@ -120,9 +120,6 @@ class ModelRunner:
            self.gpu_id, distributed=self.tp_size > 1
        )
        self.tp_group = get_tp_group()
-        self.is_multi_node_tp = not all(
-            in_the_same_node_as(self.tp_group.cpu_group, source_rank=0)
-        )

        if self.tp_size > 1:
            total_local_gpu_memory = get_available_gpu_memory(self.gpu_id)