Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
aniketmaurya committed Sep 17, 2024
1 parent d3e8cad commit 801ad63
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/litserve/server.py
Original file line number Diff line number Diff line change
Expand Up @@ -196,7 +196,7 @@ def __init__(
device_list = range(devices)
self.devices = [self.device_identifiers(accelerator, device) for device in device_list]

self.workers = self.devices * self.workers_per_device
self.inference_workers = self.devices * self.workers_per_device
self.setup_server()

def launch_inference_worker(self, num_uvicorn_servers: int):
Expand Down Expand Up @@ -414,7 +414,7 @@ def run(
sockets = [config.bind_socket()]

if num_api_servers is None:
num_api_servers = len(self.num_inference_workers)
num_api_servers = len(self.inference_workers)

if num_api_servers < 1:
raise ValueError("num_api_servers must be greater than 0")
Expand Down

0 comments on commit 801ad63

Please sign in to comment.