[misc] add zmq in collect env (vllm-project#7119)

neuralmagic · Aug 17, 2024 · cecbefe · cecbefe
1 parent 53c0dd4
commit cecbefe
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/collect_env.py b/collect_env.py
@@ -65,6 +65,7 @@
     "optree",
     "nccl",
     "transformers",
+    "zmq",
 }
 
 DEFAULT_PIP_PATTERNS = {
@@ -77,6 +78,7 @@
     "onnx",
     "nccl",
     "transformers",
+    "zmq",
 }