Improve type annotation and styles (#2926)

2025-01-16 12:51:11 -08:00
parent a883f0790d
commit bc6915e3b9
7 changed files with 78 additions and 26 deletions
--- a/python/sglang/srt/server_args.py
+++ b/python/sglang/srt/server_args.py
@@ -239,8 +239,8 @@ class ServerArgs:

        # Others
        if self.enable_dp_attention:
-            assert self.tp_size % self.dp_size == 0
            self.dp_size = self.tp_size
+            assert self.tp_size % self.dp_size == 0
            self.chunked_prefill_size = self.chunked_prefill_size // 2
            self.schedule_conservativeness = self.schedule_conservativeness * 0.3
            logger.warning(