remove tmp files

sgl-project · Nov 28, 2024 · 9f09920 · 9f09920
1 parent 4b689f2
commit 9f09920
Show file tree

Hide file tree

Showing 8 changed files with 1 addition and 958 deletions.
diff --git a/python/sglang/srt/model_executor/model_runner.py b/python/sglang/srt/model_executor/model_runner.py
@@ -472,8 +472,7 @@ def update_parameter_from_distributed(self, name, dtype, shape, empty_cache=Fals
             weights = torch.empty(shape, dtype=target_dtype, device=self.device)
 
             torch.distributed.broadcast(weights, src=0, group=self._model_update_group)
-            weights_dat = [(name, weights)]
-            self.model.load_weights(weights_dat)
+            self.model.load_weights([(name, weights)])
             if empty_cache:
                 torch.cuda.empty_cache()
 

diff --git a/test/srt/stderr.txt b/test/srt/stderr.txt
diff --git a/test/srt/stdout.txt b/test/srt/stdout.txt
diff --git a/test/srt/test.log2 b/test/srt/test.log2
diff --git a/test/srt/test_custom_process_group.py b/test/srt/test_custom_process_group.py
diff --git a/test/srt/test_custom_process_group_wraper.py b/test/srt/test_custom_process_group_wraper.py
diff --git a/test/srt/test_engine_init_parameter_update_group.py b/test/srt/test_engine_init_parameter_update_group.py