Skip to content

GitLab

Unverified Commit 3b6d539f authored by Lucius's avatar Lucius Committed by GitHub
Browse files

[Fix] Enhance DP Attention for IPv6 Compatibility (#4937)

parent 57131dd9
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
python/sglang/srt/utils.py
View file @ 3b6d539f
...@@ -942,6 +942,8 @@ def get_zmq_socket( ...@@ -942,6 +942,8 @@ def get_zmq_socket(
buf_size = -1 buf_size = -1
socket = context.socket(socket_type) socket = context.socket(socket_type)
if endpoint.find("[") != -1:
socket.setsockopt(zmq.IPV6, 1)
def set_send_opt(): def set_send_opt():
socket.setsockopt(zmq.SNDHWM, 0) socket.setsockopt(zmq.SNDHWM, 0)
......
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