Fix metrics (#15998)
This commit is contained in:
@@ -200,7 +200,7 @@ class DataParallelController:
|
||||
|
||||
self.init_dispatcher()
|
||||
|
||||
self.watchdog = Watchdog.create(
|
||||
self.soft_watchdog = Watchdog.create(
|
||||
debug_name="DataParallelController",
|
||||
watchdog_timeout=server_args.soft_watchdog_timeout,
|
||||
soft=True,
|
||||
@@ -564,7 +564,7 @@ class DataParallelController:
|
||||
def event_loop(self):
|
||||
while True:
|
||||
while True:
|
||||
self.watchdog.feed()
|
||||
self.soft_watchdog.feed()
|
||||
try:
|
||||
recv_req = self.recv_from_tokenizer.recv_pyobj(zmq.NOBLOCK)
|
||||
except zmq.ZMQError:
|
||||
|
||||
Reference in New Issue
Block a user