From f9a47cb410f99db2b17a508d7f415d2bf82dee50 Mon Sep 17 00:00:00 2001 From: FrankLeeeee Date: Tue, 4 Jul 2023 16:27:18 +0800 Subject: [PATCH] [shardformer] supported flash attention test dependency --- .github/workflows/build_on_schedule.yml | 2 +- requirements/requirements-test.txt | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/.github/workflows/build_on_schedule.yml b/.github/workflows/build_on_schedule.yml index 0589cd617b80..c38ed6b72325 100644 --- a/.github/workflows/build_on_schedule.yml +++ b/.github/workflows/build_on_schedule.yml @@ -3,7 +3,7 @@ name: Build on Schedule on: schedule: # run at 00:00 of every Sunday - - cron: '0 0 * * *' + - cron: "0 0 * * *" workflow_dispatch: jobs: diff --git a/requirements/requirements-test.txt b/requirements/requirements-test.txt index 50121a9283f2..e789d63ef7ec 100644 --- a/requirements/requirements-test.txt +++ b/requirements/requirements-test.txt @@ -16,3 +16,5 @@ triton==2.0.0.dev20221202 git+https://github.com/HazyResearch/flash-attention.git@c422fee3776eb3ea24e011ef641fd5fbeb212623#egg=flash_attn requests==2.27.1 # downgrade to avoid huggingface error https://github.com/huggingface/transformers/issues/17611 SentencePiece +ninja +git+https://github.com/facebookresearch/xformers.git@main#egg=xformers