Fix `debug_tensor_dump_output_folder` optional key missing (#4046)

87f671ab · Qubitium-ModelCloud · GitHub · 51d25405 · 87f671ab
Unverified Commit 87f671ab authored Mar 04, 2025 by Qubitium-ModelCloud Committed by GitHub Mar 04, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

python/sglang/srt/layers/logits_processor.py python/sglang/srt/layers/logits_processor.py +3 -3

No files found.
--- a/python/sglang/srt/layers/logits_processor.py
+++ b/python/sglang/srt/layers/logits_processor.py
@@ -206,9 +206,9 @@ class LogitsProcessor(nn.Module):

        from sglang.srt.managers.schedule_batch import global_server_args_dict

-        self.debug_tensor_dump_output_folder = global_server_args_dict[
-            "debug_tensor_dump_output_folder"
-        ]
+        self.debug_tensor_dump_output_folder = global_server_args_dict.get(
+            "debug_tensor_dump_output_folder", None
+        )

    def forward(
        self,