Improve sglang router (#2148)
This commit is contained in:
5
.github/workflows/release-pypi-router.yml
vendored
5
.github/workflows/release-pypi-router.yml
vendored
@@ -3,6 +3,11 @@
|
|||||||
name: Release SGLang Router to PyPI
|
name: Release SGLang Router to PyPI
|
||||||
|
|
||||||
on:
|
on:
|
||||||
|
push:
|
||||||
|
branches:
|
||||||
|
- main
|
||||||
|
paths:
|
||||||
|
- rust/pyproject.toml"
|
||||||
workflow_dispatch:
|
workflow_dispatch:
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ import argparse
|
|||||||
import copy
|
import copy
|
||||||
import multiprocessing as mp
|
import multiprocessing as mp
|
||||||
import os
|
import os
|
||||||
|
import random
|
||||||
import signal
|
import signal
|
||||||
import sys
|
import sys
|
||||||
import time
|
import time
|
||||||
@@ -11,7 +12,7 @@ import requests
|
|||||||
from sglang_router.launch_router import RouterArgs, launch_router
|
from sglang_router.launch_router import RouterArgs, launch_router
|
||||||
|
|
||||||
from sglang.srt.server import launch_server
|
from sglang.srt.server import launch_server
|
||||||
from sglang.srt.server_args import ServerArgs, prepare_server_args
|
from sglang.srt.server_args import ServerArgs
|
||||||
from sglang.srt.utils import is_port_available
|
from sglang.srt.utils import is_port_available
|
||||||
from sglang.utils import get_exception_traceback
|
from sglang.utils import get_exception_traceback
|
||||||
|
|
||||||
@@ -94,7 +95,7 @@ def find_available_ports(base_port: int, count: int) -> List[int]:
|
|||||||
while len(available_ports) < count:
|
while len(available_ports) < count:
|
||||||
if is_port_available(current_port):
|
if is_port_available(current_port):
|
||||||
available_ports.append(current_port)
|
available_ports.append(current_port)
|
||||||
current_port += 1
|
current_port += random.randint(100, 1000)
|
||||||
|
|
||||||
return available_ports
|
return available_ports
|
||||||
|
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
|
|||||||
|
|
||||||
[project]
|
[project]
|
||||||
name = "sglang-router"
|
name = "sglang-router"
|
||||||
version = "0.0.6"
|
version = "0.0.7"
|
||||||
description = "SGLang router is a standalone module implemented in Rust to achieve data parallelism across SGLang instances."
|
description = "SGLang router is a standalone module implemented in Rust to achieve data parallelism across SGLang instances."
|
||||||
authors = [{name = "Byron Hsu", email = "byronhsu1230@gmail.com"}]
|
authors = [{name = "Byron Hsu", email = "byronhsu1230@gmail.com"}]
|
||||||
requires-python = ">=3.8"
|
requires-python = ">=3.8"
|
||||||
|
|||||||
Reference in New Issue
Block a user