diff --git a/.github/workflows/nv-inference.yml b/.github/workflows/nv-inference.yml index 065f8b93f1e0..c73f279616e1 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 7255b2d76738..e4e70e6d9859 100644 --- a/requirements/requirements-triton.txt +++ b/requirements/requirements-triton.txt @@ -1 +1 @@ -triton>=2.0.0 +triton>=2.0.0,<2.1.0