Updates SGLang sampler according to their change.

PiperOrigin-RevId: 827637833

Author: The tunix Authors

tunix/generate/sglang_jax_sampler.py

@@ -108,7 +108,7 @@ def _sglang_jax_config(self, config: SglangJaxConfig):
     args["model_path"] = config.model_version
     args["precompile_bs_paddings"] = [1, 64]
     args["precompile_token_paddings"] = [8192]
-    args["disable_jax_precompile"] = True
+    args["disable_precompile"] = True
     args["page_size"] = 64
     args["context_length"] = config.context_length
     args["tp_size"] = self._find_tp_size(config.mesh)