Fix debug_tensor_dump_output_folder optional key missing (#4046)
This commit is contained in:
committed by
GitHub
parent
51d25405a7
commit
87f671ab58
@@ -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,
|
||||||
|
|||||||
Reference in New Issue
Block a user