filelock
huggingface-hub<1.0,>=0.1.0
numpy>=1.17
packaging>=20.0
pyyaml>=5.1
regex!=2019.12.17
requests
tokenizers!=0.11.3,<0.13,>=0.11.1
tqdm>=4.27

[:python_version < "3.8"]
importlib_metadata

[accelerate]
accelerate>=0.10.0

[all]
tensorflow>=2.3
onnxconverter-common
tf2onnx
tensorflow-text
torch<1.12,>=1.0
jax!=0.3.2,<=0.3.6,>=0.2.8
jaxlib<=0.3.6,>=0.1.65
flax>=0.4.1
optax>=0.0.8
sentencepiece!=0.1.92,>=0.1.91
protobuf<=3.20.1
tokenizers!=0.11.3,<0.13,>=0.11.1
torchaudio
librosa
pyctcdecode>=0.3.0
phonemizer
resampy<0.3.1
Pillow
optuna
ray[tune]
sigopt
timm
codecarbon==1.2.0
accelerate>=0.10.0

[audio]
librosa
pyctcdecode>=0.3.0
phonemizer
resampy<0.3.1

[codecarbon]
codecarbon==1.2.0

[deepspeed]
deepspeed>=0.6.5
accelerate>=0.10.0

[deepspeed-testing]
deepspeed>=0.6.5
accelerate>=0.10.0
pytest
pytest-xdist
timeout-decorator
parameterized
psutil
datasets
dill<0.3.5
pytest-timeout
black==22.3
sacrebleu<2.0.0,>=1.4.12
rouge-score
nltk
GitPython<3.1.19
hf-doc-builder>=0.3.0
protobuf<=3.20.1
sacremoses
rjieba
faiss-cpu
datasets
cookiecutter==1.7.3
optuna

[dev]
tensorflow>=2.3
onnxconverter-common
tf2onnx
tensorflow-text
torch<1.12,>=1.0
jax!=0.3.2,<=0.3.6,>=0.2.8
jaxlib<=0.3.6,>=0.1.65
flax>=0.4.1
optax>=0.0.8
sentencepiece!=0.1.92,>=0.1.91
protobuf<=3.20.1
tokenizers!=0.11.3,<0.13,>=0.11.1
torchaudio
librosa
pyctcdecode>=0.3.0
phonemizer
resampy<0.3.1
Pillow
optuna
ray[tune]
sigopt
timm
codecarbon==1.2.0
accelerate>=0.10.0
pytest
pytest-xdist
timeout-decorator
parameterized
psutil
datasets
dill<0.3.5
pytest-timeout
black==22.3
sacrebleu<2.0.0,>=1.4.12
rouge-score
nltk
GitPython<3.1.19
hf-doc-builder>=0.3.0
protobuf<=3.20.1
sacremoses
rjieba
faiss-cpu
datasets
cookiecutter==1.7.3
black==22.3
isort>=5.5.4
flake8>=3.8.3
GitPython<3.1.19
hf-doc-builder>=0.3.0
fugashi>=1.0
ipadic<2.0,>=1.0.0
unidic_lite>=1.0.7
unidic>=1.0.2
hf-doc-builder
scikit-learn
cookiecutter==1.7.3

[dev-tensorflow]
pytest
pytest-xdist
timeout-decorator
parameterized
psutil
datasets
dill<0.3.5
pytest-timeout
black==22.3
sacrebleu<2.0.0,>=1.4.12
rouge-score
nltk
GitPython<3.1.19
hf-doc-builder>=0.3.0
protobuf<=3.20.1
sacremoses
rjieba
faiss-cpu
datasets
cookiecutter==1.7.3
tensorflow>=2.3
onnxconverter-common
tf2onnx
tensorflow-text
sentencepiece!=0.1.92,>=0.1.91
protobuf<=3.20.1
tokenizers!=0.11.3,<0.13,>=0.11.1
Pillow
black==22.3
isort>=5.5.4
flake8>=3.8.3
GitPython<3.1.19
hf-doc-builder>=0.3.0
hf-doc-builder
scikit-learn
cookiecutter==1.7.3
onnxconverter-common
tf2onnx
onnxruntime>=1.4.0
onnxruntime-tools>=1.4.2
librosa
pyctcdecode>=0.3.0
phonemizer
resampy<0.3.1

[dev-torch]
pytest
pytest-xdist
timeout-decorator
parameterized
psutil
datasets
dill<0.3.5
pytest-timeout
black==22.3
sacrebleu<2.0.0,>=1.4.12
rouge-score
nltk
GitPython<3.1.19
hf-doc-builder>=0.3.0
protobuf<=3.20.1
sacremoses
rjieba
faiss-cpu
datasets
cookiecutter==1.7.3
torch<1.12,>=1.0
sentencepiece!=0.1.92,>=0.1.91
protobuf<=3.20.1
tokenizers!=0.11.3,<0.13,>=0.11.1
torchaudio
librosa
pyctcdecode>=0.3.0
phonemizer
resampy<0.3.1
Pillow
optuna
ray[tune]
sigopt
timm
codecarbon==1.2.0
black==22.3
isort>=5.5.4
flake8>=3.8.3
GitPython<3.1.19
hf-doc-builder>=0.3.0
fugashi>=1.0
ipadic<2.0,>=1.0.0
unidic_lite>=1.0.7
unidic>=1.0.2
hf-doc-builder
scikit-learn
cookiecutter==1.7.3
onnxruntime>=1.4.0
onnxruntime-tools>=1.4.2

[docs]
tensorflow>=2.3
onnxconverter-common
tf2onnx
tensorflow-text
torch<1.12,>=1.0
jax!=0.3.2,<=0.3.6,>=0.2.8
jaxlib<=0.3.6,>=0.1.65
flax>=0.4.1
optax>=0.0.8
sentencepiece!=0.1.92,>=0.1.91
protobuf<=3.20.1
tokenizers!=0.11.3,<0.13,>=0.11.1
torchaudio
librosa
pyctcdecode>=0.3.0
phonemizer
resampy<0.3.1
Pillow
optuna
ray[tune]
sigopt
timm
codecarbon==1.2.0
accelerate>=0.10.0
hf-doc-builder

[docs_specific]
hf-doc-builder

[fairscale]
fairscale>0.3

[flax]
jax!=0.3.2,<=0.3.6,>=0.2.8
jaxlib<=0.3.6,>=0.1.65
flax>=0.4.1
optax>=0.0.8

[flax-speech]
librosa
pyctcdecode>=0.3.0
phonemizer
resampy<0.3.1

[ftfy]
ftfy

[integrations]
optuna
ray[tune]
sigopt

[ja]
fugashi>=1.0
ipadic<2.0,>=1.0.0
unidic_lite>=1.0.7
unidic>=1.0.2

[modelcreation]
cookiecutter==1.7.3

[onnx]
onnxconverter-common
tf2onnx
onnxruntime>=1.4.0
onnxruntime-tools>=1.4.2

[onnxruntime]
onnxruntime>=1.4.0
onnxruntime-tools>=1.4.2

[optuna]
optuna

[quality]
black==22.3
isort>=5.5.4
flake8>=3.8.3
GitPython<3.1.19
hf-doc-builder>=0.3.0

[ray]
ray[tune]

[retrieval]
faiss-cpu
datasets

[sagemaker]
sagemaker>=2.31.0

[sentencepiece]
sentencepiece!=0.1.92,>=0.1.91
protobuf<=3.20.1

[serving]
pydantic
uvicorn
fastapi
starlette

[sigopt]
sigopt

[sklearn]
scikit-learn

[speech]
torchaudio
librosa
pyctcdecode>=0.3.0
phonemizer
resampy<0.3.1

[testing]
pytest
pytest-xdist
timeout-decorator
parameterized
psutil
datasets
dill<0.3.5
pytest-timeout
black==22.3
sacrebleu<2.0.0,>=1.4.12
rouge-score
nltk
GitPython<3.1.19
hf-doc-builder>=0.3.0
protobuf<=3.20.1
sacremoses
rjieba
faiss-cpu
datasets
cookiecutter==1.7.3

[tf]
tensorflow>=2.3
onnxconverter-common
tf2onnx
tensorflow-text

[tf-cpu]
tensorflow-cpu>=2.3
onnxconverter-common
tf2onnx
tensorflow-text

[tf-speech]
librosa
pyctcdecode>=0.3.0
phonemizer
resampy<0.3.1

[timm]
timm

[tokenizers]
tokenizers!=0.11.3,<0.13,>=0.11.1

[torch]
torch<1.12,>=1.0

[torch-speech]
torchaudio
librosa
pyctcdecode>=0.3.0
phonemizer
resampy<0.3.1

[torchhub]
filelock
huggingface-hub<1.0,>=0.1.0
importlib_metadata
numpy>=1.17
packaging>=20.0
protobuf<=3.20.1
regex!=2019.12.17
requests
sentencepiece!=0.1.92,>=0.1.91
torch<1.12,>=1.0
tokenizers!=0.11.3,<0.13,>=0.11.1
tqdm>=4.27

[vision]
Pillow
