Unverified Commit 87f671ab authored by Qubitium-ModelCloud's avatar Qubitium-ModelCloud Committed by GitHub
Browse files

Fix `debug_tensor_dump_output_folder` optional key missing (#4046)

parent 51d25405
...@@ -206,9 +206,9 @@ class LogitsProcessor(nn.Module): ...@@ -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,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment