reverting renaming to be consistent with other workers

Signed-off-by: Yannick Schnider <[email protected]>
IBM · Jan 27, 2025 · bdde43e · bdde43e
1 parent beadbd5
commit bdde43e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vllm/worker/spyre_worker.py b/vllm/worker/spyre_worker.py
@@ -59,7 +59,7 @@ def __init__(
                                                  self.is_driver_worker)
         self._env_initialized = False
 
-    def init_distributed_spyre_environment(self) -> None:
+    def init_distributed_environment(self) -> None:
         """Initialize the distributed environment."""
 
         torch._C._distributed_c10d._register_process_group(
@@ -91,7 +91,7 @@ def init_device(self) -> None:
             )
 
             if self.parallel_config.world_size > 1:
-                self.init_distributed_spyre_environment()
+                self.init_distributed_environment()
             elif envs.VLLM_SPYRE_DYNAMO_BACKEND in [
                     "sendnn", "sendnn_decoder"
             ]: