Rename arguments --disable-nan-detection to --enable-nan-detection (#2066)

2024-11-17 16:53:44 -08:00
parent 62832bb272
commit ebaa2f3199
5 changed files with 17 additions and 20 deletions
--- a/python/sglang/srt/layers/sampler.py
+++ b/python/sglang/srt/layers/sampler.py
@@ -29,7 +29,7 @@ logger = logging.getLogger(__name__)
 class Sampler(nn.Module):
    def __init__(self):
        super().__init__()
-        self.use_nan_detectioin = not global_server_args_dict["disable_nan_detection"]
+        self.use_nan_detectioin = global_server_args_dict["enable_nan_detection"]

    def forward(
        self,