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

650a4cc5 · Michael Goin · GitHub · 9ca62d86 · 650a4cc5
Unverified Commit 650a4cc5 authored Jun 04, 2024 by Michael Goin Committed by GitHub Jun 04, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

collect_env.py collect_env.py +2 -0

No files found.
--- 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",
 }