From f96c1c0a783bc621bb607ec5b6650c804abfe7df Mon Sep 17 00:00:00 2001 From: Lev Kurilenko <113481193+lekurile@users.noreply.github.com> Date: Fri, 1 Sep 2023 16:47:52 -0700 Subject: [PATCH] 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 --- .github/workflows/nv-inference.yml | 1 + requirements/requirements-triton.txt | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/nv-inference.yml b/.github/workflows/nv-inference.yml index 065f8b93..c73f2796 100644 --- 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 index 7255b2d7..e4e70e6d 100644 --- a/requirements/requirements-triton.txt +++ b/requirements/requirements-triton.txt @@ -1 +1 @@ -triton>=2.0.0 +triton>=2.0.0,<2.1.0 -- GitLab