[server] Passing model_override_args to launch_server via the CLI. (#1298)

Signed-off-by: Kai-Hsun Chen <kaihsun@anyscale.com>
This commit is contained in:
Kai-Hsun Chen
2024-09-09 02:14:25 -07:00
committed by GitHub
parent 662ecd9368
commit c9b75917d5
8 changed files with 71 additions and 16 deletions

View File

@@ -0,0 +1,24 @@
import unittest
from sglang.srt.server_args import prepare_server_args
class TestPrepareServerArgs(unittest.TestCase):
def test_prepare_server_args(self):
server_args = prepare_server_args(
[
"--model-path",
"model_path",
"--json-model-override-args",
'{"rope_scaling": {"factor": 2.0, "type": "linear"}}',
]
)
self.assertEqual(server_args.model_path, "model_path")
self.assertEqual(
server_args.json_model_override_args,
{"rope_scaling": {"factor": 2.0, "type": "linear"}},
)
if __name__ == "__main__":
unittest.main()