Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "be79cd7d8e3911d67230443eef7faf0f486533db"
Unverified Commit 4b6aecb4 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Pin numba for now (#23118)

parent 3ff89f29
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
setup.py
View file @ 4b6aecb4
...@@ -135,6 +135,7 @@ _deps = [ ...@@ -135,6 +135,7 @@ _deps = [
"librosa", "librosa",
"nltk", "nltk",
"natten>=0.14.6", "natten>=0.14.6",
"numba<0.57.0", # Can be removed once unpinned.
"numpy>=1.17", "numpy>=1.17",
"onnxconverter-common", "onnxconverter-common",
"onnxruntime-tools>=1.4.2", "onnxruntime-tools>=1.4.2",
...@@ -286,7 +287,8 @@ extras["sigopt"] = deps_list("sigopt") ...@@ -286,7 +287,8 @@ extras["sigopt"] = deps_list("sigopt")
extras["integrations"] = extras["optuna"] + extras["ray"] + extras["sigopt"] extras["integrations"] = extras["optuna"] + extras["ray"] + extras["sigopt"]
extras["serving"] = deps_list("pydantic", "uvicorn", "fastapi", "starlette") extras["serving"] = deps_list("pydantic", "uvicorn", "fastapi", "starlette")
extras["audio"] = deps_list("librosa", "pyctcdecode", "phonemizer", "kenlm") # numba can be removed here once unpinned
extras["audio"] = deps_list("librosa", "pyctcdecode", "phonemizer", "kenlm", "numba")
# `pip install ".[speech]"` is deprecated and `pip install ".[torch-speech]"` should be used instead # `pip install ".[speech]"` is deprecated and `pip install ".[torch-speech]"` should be used instead
extras["speech"] = deps_list("torchaudio") + extras["audio"] extras["speech"] = deps_list("torchaudio") + extras["audio"]
extras["torch-speech"] = deps_list("torchaudio") + extras["audio"] extras["torch-speech"] = deps_list("torchaudio") + extras["audio"]
......
src/transformers/dependency_versions_table.py
View file @ 4b6aecb4
...@@ -36,6 +36,7 @@ deps = { ...@@ -36,6 +36,7 @@ deps = {
"librosa": "librosa", "librosa": "librosa",
"nltk": "nltk", "nltk": "nltk",
"natten": "natten>=0.14.6", "natten": "natten>=0.14.6",
"numba": "numba<0.57.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",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment