CI: limit natten version (#28432)
This commit is contained in:
@@ -470,7 +470,7 @@ exotic_models_job = CircleCIJob(
|
|||||||
"pip install -U --upgrade-strategy eager 'git+https://github.com/facebookresearch/detectron2.git'",
|
"pip install -U --upgrade-strategy eager 'git+https://github.com/facebookresearch/detectron2.git'",
|
||||||
"sudo apt install tesseract-ocr",
|
"sudo apt install tesseract-ocr",
|
||||||
"pip install -U --upgrade-strategy eager pytesseract",
|
"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 python-Levenshtein",
|
||||||
"pip install -U --upgrade-strategy eager opencv-python",
|
"pip install -U --upgrade-strategy eager opencv-python",
|
||||||
"pip install -U --upgrade-strategy eager nltk",
|
"pip install -U --upgrade-strategy eager nltk",
|
||||||
@@ -514,7 +514,7 @@ doc_test_job = CircleCIJob(
|
|||||||
"pip install -U --upgrade-strategy eager -e .[dev]",
|
"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 -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 --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",
|
"pip install -U --upgrade-strategy eager g2p-en",
|
||||||
"find -name __pycache__ -delete",
|
"find -name __pycache__ -delete",
|
||||||
"find . -name \*.pyc -delete",
|
"find . -name \*.pyc -delete",
|
||||||
|
|||||||
2
.github/workflows/check_tiny_models.yml
vendored
2
.github/workflows/check_tiny_models.yml
vendored
@@ -36,7 +36,7 @@ jobs:
|
|||||||
pip install --upgrade pip
|
pip install --upgrade pip
|
||||||
python -m pip install -U .[sklearn,torch,testing,sentencepiece,torch-speech,vision,timm,video,tf-cpu]
|
python -m pip install -U .[sklearn,torch,testing,sentencepiece,torch-speech,vision,timm,video,tf-cpu]
|
||||||
pip install tensorflow_probability
|
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)
|
- name: Create all tiny models (locally)
|
||||||
run: |
|
run: |
|
||||||
|
|||||||
@@ -65,7 +65,7 @@ RUN python3 -m pip install --no-cache-dir git+https://github.com/huggingface/opt
|
|||||||
RUN python3 -m pip install --no-cache-dir decord av==9.2.0
|
RUN python3 -m pip install --no-cache-dir decord av==9.2.0
|
||||||
|
|
||||||
# For `dinat` model
|
# 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
|
# For `nougat` tokenizer
|
||||||
RUN python3 -m pip install --no-cache-dir python-Levenshtein
|
RUN python3 -m pip install --no-cache-dir python-Levenshtein
|
||||||
|
|||||||
2
setup.py
2
setup.py
@@ -130,7 +130,7 @@ _deps = [
|
|||||||
"keras-nlp>=0.3.1",
|
"keras-nlp>=0.3.1",
|
||||||
"librosa",
|
"librosa",
|
||||||
"nltk",
|
"nltk",
|
||||||
"natten>=0.14.6",
|
"natten>=0.14.6,<0.15.0",
|
||||||
"numpy>=1.17",
|
"numpy>=1.17",
|
||||||
"onnxconverter-common",
|
"onnxconverter-common",
|
||||||
"onnxruntime-tools>=1.4.2",
|
"onnxruntime-tools>=1.4.2",
|
||||||
|
|||||||
@@ -36,7 +36,7 @@ deps = {
|
|||||||
"keras-nlp": "keras-nlp>=0.3.1",
|
"keras-nlp": "keras-nlp>=0.3.1",
|
||||||
"librosa": "librosa",
|
"librosa": "librosa",
|
||||||
"nltk": "nltk",
|
"nltk": "nltk",
|
||||||
"natten": "natten>=0.14.6",
|
"natten": "natten>=0.14.6,<0.15.0",
|
||||||
"numpy": "numpy>=1.17",
|
"numpy": "numpy>=1.17",
|
||||||
"onnxconverter-common": "onnxconverter-common",
|
"onnxconverter-common": "onnxconverter-common",
|
||||||
"onnxruntime-tools": "onnxruntime-tools>=1.4.2",
|
"onnxruntime-tools": "onnxruntime-tools>=1.4.2",
|
||||||
|
|||||||
Reference in New Issue
Block a user