EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Fix debug_tensor_dump_output_folder optional key missing (#4046)

Browse Source
This commit is contained in:
Qubitium-ModelCloud
2025-03-04 19:42:48 +08:00
committed by GitHub
parent 51d25405a7
commit 87f671ab58
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/sglang/srt/layers/logits_processor.py
View File

@@ -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,