diff --git a/.circleci/create_circleci_config.py b/.circleci/create_circleci_config.py index 8fd237a5ae7471..e0c913cb364c24 100644 --- a/.circleci/create_circleci_config.py +++ b/.circleci/create_circleci_config.py @@ -470,7 +470,7 @@ def job_name(self): "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", + "pip install -U --upgrade-strategy eager 'natten<0.15.0'", "pip install -U --upgrade-strategy eager python-Levenshtein", "pip install -U --upgrade-strategy eager opencv-python", "pip install -U --upgrade-strategy eager nltk", @@ -514,7 +514,7 @@ def job_name(self): "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 pytest-sugar", - "pip install -U --upgrade-strategy eager natten", + "pip install -U --upgrade-strategy eager 'natten<0.15.0'", "pip install -U --upgrade-strategy eager g2p-en", "find -name __pycache__ -delete", "find . -name \*.pyc -delete", diff --git a/.github/workflows/check_tiny_models.yml b/.github/workflows/check_tiny_models.yml index 898e441a4234c6..0725bd04a1f2c3 100644 --- a/.github/workflows/check_tiny_models.yml +++ b/.github/workflows/check_tiny_models.yml @@ -36,7 +36,7 @@ jobs: pip install --upgrade pip python -m pip install -U .[sklearn,torch,testing,sentencepiece,torch-speech,vision,timm,video,tf-cpu] pip install tensorflow_probability - python -m pip install -U natten + python -m pip install -U 'natten<0.15.0' - name: Create all tiny models (locally) run: | @@ -62,7 +62,7 @@ jobs: path: reports/tests_pipelines - name: Create + Upload tiny models for new model architecture(s) - run: | + run: | python utils/update_tiny_models.py --num_workers 2 - name: Full report diff --git a/docker/transformers-all-latest-gpu/Dockerfile b/docker/transformers-all-latest-gpu/Dockerfile index baa0804430c253..96525e247f60c1 100644 --- a/docker/transformers-all-latest-gpu/Dockerfile +++ b/docker/transformers-all-latest-gpu/Dockerfile @@ -58,14 +58,14 @@ RUN python3 -m pip install --no-cache-dir einops # Add autoawq for quantization testing RUN python3 -m pip install --no-cache-dir https://github.com/casper-hansen/AutoAWQ/releases/download/v0.1.8/autoawq-0.1.8+cu118-cp38-cp38-linux_x86_64.whl -# For bettertransformer + gptq +# For bettertransformer + gptq RUN python3 -m pip install --no-cache-dir git+https://github.com/huggingface/optimum@main#egg=optimum # For video model testing RUN python3 -m pip install --no-cache-dir decord av==9.2.0 # For `dinat` model -RUN python3 -m pip install --no-cache-dir natten -f https://shi-labs.com/natten/wheels/$CUDA/ +RUN python3 -m pip install --no-cache-dir natten<0.15.0 -f https://shi-labs.com/natten/wheels/$CUDA/ # For `nougat` tokenizer RUN python3 -m pip install --no-cache-dir python-Levenshtein diff --git a/setup.py b/setup.py index aba442ff42ba25..1f434df347b620 100644 --- a/setup.py +++ b/setup.py @@ -130,7 +130,7 @@ "keras-nlp>=0.3.1", "librosa", "nltk", - "natten>=0.14.6", + "natten>=0.14.6,<0.15.0", "numpy>=1.17", "onnxconverter-common", "onnxruntime-tools>=1.4.2", diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index fcace1826ac453..f3b811c2d181f1 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -36,7 +36,7 @@ "keras-nlp": "keras-nlp>=0.3.1", "librosa": "librosa", "nltk": "nltk", - "natten": "natten>=0.14.6", + "natten": "natten>=0.14.6,<0.15.0", "numpy": "numpy>=1.17", "onnxconverter-common": "onnxconverter-common", "onnxruntime-tools": "onnxruntime-tools>=1.4.2",