Skip to content

GitLab

Unverified Commit 5495883c authored by Kirthi Shankar Sivamani's avatar Kirthi Shankar Sivamani Committed by GitHub
Browse files

Pin FA version (#254) 



pin FA version
Signed-off-by: default avatarKirthi Shankar Sivamani <ksivamani@nvidia.com>
parent 37bbfc76
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
setup.py
View file @ 5495883c
...@@ -283,7 +283,7 @@ def setup_requirements() -> Tuple[List[str], List[str], List[str]]: ...@@ -283,7 +283,7 @@ def setup_requirements() -> Tuple[List[str], List[str], List[str]]:
# Framework-specific requirements # Framework-specific requirements
if "pytorch" in frameworks(): if "pytorch" in frameworks():
add_unique(install_reqs, ["torch", "flash-attn>=1.0.2"]) add_unique(install_reqs, ["torch", "flash-attn==1.0.6"])
add_unique(test_reqs, ["numpy", "onnxruntime", "torchvision"]) add_unique(test_reqs, ["numpy", "onnxruntime", "torchvision"])
if "jax" in frameworks(): if "jax" in frameworks():
if not found_pybind11(): if not found_pybind11():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment