From 176fa4bc6ed7a5377809122e69577d7adaaab901 Mon Sep 17 00:00:00 2001 From: Jensen Date: Wed, 4 Feb 2026 14:34:15 +0800 Subject: [PATCH] [chore] Remove zmq context initialize (#1187) Signed-off-by: xiedeyantu --- vllm_omni/diffusion/scheduler.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/vllm_omni/diffusion/scheduler.py b/vllm_omni/diffusion/scheduler.py index f104d0522..bdeabd998 100644 --- a/vllm_omni/diffusion/scheduler.py +++ b/vllm_omni/diffusion/scheduler.py @@ -13,14 +13,13 @@ logger = init_logger(__name__) class Scheduler: def initialize(self, od_config: OmniDiffusionConfig): - existing_context = getattr(self, "context", None) - if existing_context is not None and not existing_context.closed: + existing_mq = getattr(self, "mq", None) + if existing_mq is not None and not existing_mq.closed: logger.warning("SyncSchedulerClient is already initialized. Re-initializing.") self.close() self.num_workers = od_config.num_gpus self.od_config = od_config - self.context = zmq.Context() # Standard synchronous context # Initialize single MessageQueue for all message types (generation & RPC) # Assuming all readers are local for now as per current launch_engine implementation @@ -72,8 +71,5 @@ class Scheduler: def close(self): """Closes the socket and terminates the context.""" - if hasattr(self, "context"): - self.context.term() - self.context = None self.mq = None self.result_mq = None