Fix debug_tensor_dump_output_folder optional key missing (#4046)

This commit is contained in:
Qubitium-ModelCloud
2025-03-04 19:42:48 +08:00
committed by GitHub
parent 51d25405a7
commit 87f671ab58

View File

@@ -206,9 +206,9 @@ class LogitsProcessor(nn.Module):
from sglang.srt.managers.schedule_batch import global_server_args_dict from sglang.srt.managers.schedule_batch import global_server_args_dict
self.debug_tensor_dump_output_folder = global_server_args_dict[ self.debug_tensor_dump_output_folder = global_server_args_dict.get(
"debug_tensor_dump_output_folder" "debug_tensor_dump_output_folder", None
] )
def forward( def forward(
self, self,