Pin Triton version to >=2.0.0 and <2.1.0 (#4251)

* Pin Triton version to 2.0.0 * Pin Triton version to < 2.1.0 * Add >=2.0.0 * pin transformers version --------- Co-authored-by: Michael Wyatt <[email protected]>
deepspeedai · Sep 1, 2023 · f96c1c0 · f96c1c0
1 parent 844eb68
commit f96c1c0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.github/workflows/nv-inference.yml b/.github/workflows/nv-inference.yml
@@ -34,6 +34,7 @@ jobs:
         run: |
           git clone https://github.com/huggingface/transformers
           cd transformers
+          git checkout ccb92be
           git rev-parse --short HEAD
           pip install .
 

diff --git a/requirements/requirements-triton.txt b/requirements/requirements-triton.txt
@@ -1 +1 @@
-triton>=2.0.0
+triton>=2.0.0,<2.1.0