diff --git a/collect_env.py b/collect_env.py index 1ecfeb8e..5a015c3c 100644 --- a/collect_env.py +++ b/collect_env.py @@ -64,6 +64,7 @@ DEFAULT_CONDA_PATTERNS = { "triton", "optree", "nccl", + "transformers", } DEFAULT_PIP_PATTERNS = { @@ -75,6 +76,7 @@ DEFAULT_PIP_PATTERNS = { "optree", "onnx", "nccl", + "transformers", }