From 0e75aeefaf4beeb7a5bb6a1f05b83ab99e045a24 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Fri, 2 Feb 2024 13:11:48 +0100 Subject: [PATCH] Fix issues caused by natten (#28834) try Co-authored-by: ydshieh --- .circleci/create_circleci_config.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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",