From c9a61e65713c1de1cb422c086913a819cf9f4ed3 Mon Sep 17 00:00:00 2001 From: muellerzr Date: Wed, 4 May 2022 09:22:46 -0400 Subject: [PATCH] Update to build via git for accelerate --- .circleci/config.yml | 2 -- examples/pytorch/_tests_requirements.txt | 2 +- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index 57a0e7d31337..5cf2fab47635 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -587,7 +587,6 @@ jobs: - run: pip install --upgrade pip - run: pip install .[sklearn,torch,sentencepiece,testing,torch-speech] - run: pip install -r examples/pytorch/_tests_requirements.txt - - run: pip install git+https://github.com/huggingface/accelerate - save_cache: key: v0.4-torch_examples-{{ checksum "setup.py" }} paths: @@ -623,7 +622,6 @@ jobs: - run: pip install --upgrade pip - run: pip install .[sklearn,torch,sentencepiece,testing,torch-speech] - run: pip install -r examples/pytorch/_tests_requirements.txt - - run: pip install git+https://github.com/huggingface/accelerate - save_cache: key: v0.4-torch_examples-{{ checksum "setup.py" }} paths: diff --git a/examples/pytorch/_tests_requirements.txt b/examples/pytorch/_tests_requirements.txt index 9483d3a750a3..8c13e79aa44b 100644 --- a/examples/pytorch/_tests_requirements.txt +++ b/examples/pytorch/_tests_requirements.txt @@ -3,7 +3,7 @@ scikit-learn seqeval psutil sacrebleu >= 1.4.12 -accelerate >= 0.5.0 +git+https://github.com/huggingface/accelerate@main#egg=accelerate rouge-score tensorflow_datasets matplotlib