diff --git a/.circleci/create_circleci_config.py b/.circleci/create_circleci_config.py index 87b6a1342c..16b39a4dae 100644 --- a/.circleci/create_circleci_config.py +++ b/.circleci/create_circleci_config.py @@ -470,7 +470,7 @@ exotic_models_job = CircleCIJob( "pip install -U --upgrade-strategy eager 'git+https://github.com/facebookresearch/detectron2.git'", "sudo apt install tesseract-ocr", "pip install -U --upgrade-strategy eager pytesseract", - "pip install -U --upgrade-strategy eager 'natten<0.15.0'", + "pip install -U --upgrade-strategy eager natten==0.15.1+torch210cpu -f https://shi-labs.com/natten/wheels", "pip install -U --upgrade-strategy eager python-Levenshtein", "pip install -U --upgrade-strategy eager opencv-python", "pip install -U --upgrade-strategy eager nltk", @@ -515,7 +515,7 @@ doc_test_job = CircleCIJob( "pip install -U --upgrade-strategy eager -e .[dev]", "pip install -U --upgrade-strategy eager -e git+https://github.com/huggingface/accelerate@main#egg=accelerate", "pip install --upgrade --upgrade-strategy eager 'pytest<8.0.0' pytest-sugar", - "pip install -U --upgrade-strategy eager 'natten<0.15.0'", + "pip install -U --upgrade-strategy eager natten==0.15.1+torch210cpu -f https://shi-labs.com/natten/wheels", "pip install -U --upgrade-strategy eager g2p-en", "find -name __pycache__ -delete", "find . -name \*.pyc -delete",