Skip to content
Prev Previous commit
Next Next commit
fix func call
  • Loading branch information
sniper35 committed Feb 8, 2026
commit a3c34a52012837f203a47b326f09a3b37c5dadf5
3 changes: 2 additions & 1 deletion examples/diffusion_router/demo.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,8 @@ def main():

# Pre-register any workers specified on the command line
for url in args.worker_urls:
router.add_worker_sync(url)
router.worker_request_counts[url] = 0
router.worker_failure_counts[url] = 0
if args.verbose:
print(f"[demo] Pre-registered worker: {url}")

Expand Down