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