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

2025-04-21 15:44:11 +08:00
parent 57131dd955
commit 3b6d539f63
1 changed files with 2 additions and 0 deletions
--- a/python/sglang/srt/utils.py
+++ b/python/sglang/srt/utils.py
@@ -942,6 +942,8 @@ def get_zmq_socket(
        buf_size = -1

    socket = context.socket(socket_type)
+    if endpoint.find("[") != -1:
+        socket.setsockopt(zmq.IPV6, 1)

    def set_send_opt():
        socket.setsockopt(zmq.SNDHWM, 0)