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