Skip to content

GitLab

Unverified Commit 5797fb97 authored by Rui Qiao's avatar Rui Qiao Committed by GitHub
Browse files

[Misc] Remove ignore_reinit_error for ray.init() (#15373)

parent 3892e58a
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 5 deletions
vllm/executor/ray_utils.py
View file @ 5797fb97
...@@ -289,16 +289,14 @@ def initialize_ray_cluster( ...@@ -289,16 +289,14 @@ def initialize_ray_cluster(
elif current_platform.is_rocm() or current_platform.is_xpu(): elif current_platform.is_rocm() or current_platform.is_xpu():
# Try to connect existing ray instance and create a new one if not found # Try to connect existing ray instance and create a new one if not found
try: try:
ray.init("auto", ignore_reinit_error=True) ray.init("auto")
except ConnectionError: except ConnectionError:
logger.warning( logger.warning(
"No existing RAY instance detected. " "No existing RAY instance detected. "
"A new instance will be launched with current node resources.") "A new instance will be launched with current node resources.")
ray.init(address=ray_address, ray.init(address=ray_address, num_gpus=parallel_config.world_size)
ignore_reinit_error=True,
num_gpus=parallel_config.world_size)
else: else:
ray.init(address=ray_address, ignore_reinit_error=True) ray.init(address=ray_address)
device_str = current_platform.ray_device_key device_str = current_platform.ray_device_key
if not device_str: if not device_str:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment