[Misc] Add transformers version to collect_env.py (#5259)

2024-06-04 15:52:28 -04:00
parent 9ca62d8668
commit 650a4cc55e
1 changed files with 2 additions and 0 deletions
--- 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",
 }