From 87f671ab584e256cbca758466ad391be15a11c54 Mon Sep 17 00:00:00 2001 From: Qubitium-ModelCloud Date: Tue, 4 Mar 2025 19:42:48 +0800 Subject: [PATCH] Fix `debug_tensor_dump_output_folder` optional key missing (#4046) --- python/sglang/srt/layers/logits_processor.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/sglang/srt/layers/logits_processor.py b/python/sglang/srt/layers/logits_processor.py index 59c6daa31..4996ae78e 100644 --- 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,