[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:
24
test/srt/test_server_args.py
Normal file
24
test/srt/test_server_args.py
Normal 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()
|
||||
Reference in New Issue
Block a user