diff --git a/collect_env.py b/collect_env.py index 083cb768..244e4ddd 100644 --- a/collect_env.py +++ b/collect_env.py @@ -65,6 +65,7 @@ DEFAULT_CONDA_PATTERNS = { "optree", "nccl", "transformers", + "zmq", } DEFAULT_PIP_PATTERNS = { @@ -77,6 +78,7 @@ DEFAULT_PIP_PATTERNS = { "onnx", "nccl", "transformers", + "zmq", }