Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
7eebe8cc
Unverified
Commit
7eebe8cc
authored
Aug 21, 2024
by
youkaichao
Committed by
GitHub
Aug 21, 2024
Browse files
[distributed][misc] error on same VLLM_HOST_IP setting (#7756)
parent
8678a69a
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
17 additions
and
1 deletion
+17
-1
vllm/envs.py
vllm/envs.py
+4
-1
vllm/executor/ray_gpu_executor.py
vllm/executor/ray_gpu_executor.py
+13
-0
No files found.
vllm/envs.py
View file @
7eebe8cc
...
...
@@ -137,7 +137,10 @@ environment_variables: Dict[str, Callable[[], Any]] = {
os
.
path
.
join
(
get_default_cache_root
(),
"vllm"
),
)),
# used in distributed environment to determine the master address
# used in distributed environment to determine the ip address
# of the current node, when the node has multiple network interfaces.
# If you are using multi-node inference, you should set this differently
# on each node.
'VLLM_HOST_IP'
:
lambda
:
os
.
getenv
(
'VLLM_HOST_IP'
,
""
)
or
os
.
getenv
(
"HOST_IP"
,
""
),
...
...
vllm/executor/ray_gpu_executor.py
View file @
7eebe8cc
...
...
@@ -218,6 +218,19 @@ class RayGPUExecutor(DistributedGPUExecutor):
for
node_id
,
gpu_ids
in
node_gpus
.
items
():
node_gpus
[
node_id
]
=
sorted
(
gpu_ids
)
all_ips
=
set
(
worker_ips
+
[
driver_ip
])
n_ips
=
len
(
all_ips
)
n_nodes
=
len
(
node_workers
)
if
n_nodes
!=
n_ips
:
raise
RuntimeError
(
f
"Every node should have a unique IP address. Got
{
n_nodes
}
"
f
" nodes with node ids
{
list
(
node_workers
.
keys
())
}
and "
f
"
{
n_ips
}
unique IP addresses
{
all_ips
}
. Please check your"
" network configuration. If you set `VLLM_HOST_IP` or "
"`HOST_IP` environment variable, make sure it is unique for"
" each node."
)
VLLM_INSTANCE_ID
=
get_vllm_instance_id
()
# Set environment variables for the driver and workers.
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment