From 18c6b2123092ba0cb736d1968a3fb766fd5eee5c Mon Sep 17 00:00:00 2001 From: KKZ20 Date: Wed, 27 Dec 2023 17:38:55 +0800 Subject: [PATCH 1/2] doc/update requirements-test.txt --- requirements/requirements-test.txt | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/requirements/requirements-test.txt b/requirements/requirements-test.txt index 61b58055e666..4136cefc3c37 100644 --- a/requirements/requirements-test.txt +++ b/requirements/requirements-test.txt @@ -1,5 +1,4 @@ diffusers -fbgemm-gpu==0.2.0 pytest coverage==7.2.3 git+https://github.com/hpcaitech/pytest-testmon @@ -16,7 +15,7 @@ triton==2.1.0 requests==2.27.1 # downgrade to avoid huggingface error https://github.com/huggingface/transformers/issues/17611 SentencePiece ninja -flash_attn==2.0.5 +flash_attn datasets pydantic ray From f559e2c65344ae177a56f6519d60801a44d94439 Mon Sep 17 00:00:00 2001 From: KKZ20 Date: Wed, 27 Dec 2023 17:59:59 +0800 Subject: [PATCH 2/2] update torch-cuda compatibility check --- .compatibility | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/.compatibility b/.compatibility index 32da32be5521..a918cb162216 100644 --- a/.compatibility +++ b/.compatibility @@ -1,3 +1,2 @@ -1.12.0-11.3.0 -1.13.0-11.6.0 2.0.0-11.7.0 +2.1.0-11.8.0