[Misc] collect flashinfer version in collect_env.py (#24378)
Signed-off-by: Ye (Charlotte) Qi <yeq@meta.com>
This commit is contained in:
committed by
GitHub
parent
75334956c2
commit
81c53ef55c
@@ -74,6 +74,7 @@ DEFAULT_CONDA_PATTERNS = {
|
|||||||
"zmq",
|
"zmq",
|
||||||
"nvidia",
|
"nvidia",
|
||||||
"pynvml",
|
"pynvml",
|
||||||
|
"flashinfer-python",
|
||||||
}
|
}
|
||||||
|
|
||||||
DEFAULT_PIP_PATTERNS = {
|
DEFAULT_PIP_PATTERNS = {
|
||||||
@@ -89,6 +90,7 @@ DEFAULT_PIP_PATTERNS = {
|
|||||||
"zmq",
|
"zmq",
|
||||||
"nvidia",
|
"nvidia",
|
||||||
"pynvml",
|
"pynvml",
|
||||||
|
"flashinfer-python",
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user