dependency_versions_table.py 3.13 KB
Newer Older
1
2
3
4
# THIS FILE HAS BEEN AUTOGENERATED. To update:
# 1. modify the `_deps` dict in setup.py
# 2. run `make deps_table_update``
deps = {
Yih-Dar's avatar
Yih-Dar committed
5
    "Pillow": "Pillow<10.0.0",
6
    "accelerate": "accelerate>=0.20.3",
7
    "av": "av==9.2.0",
Sylvain Gugger's avatar
Sylvain Gugger committed
8
    "beautifulsoup4": "beautifulsoup4",
9
    "black": "black~=23.1",
10
    "codecarbon": "codecarbon==1.2.0",
Lysandre Debut's avatar
Lysandre Debut committed
11
    "cookiecutter": "cookiecutter==1.7.3",
12
    "dataclasses": "dataclasses",
13
    "datasets": "datasets!=2.5.0",
14
    "decord": "decord==0.6.0",
15
    "deepspeed": "deepspeed>=0.9.3",
Lysandre Debut's avatar
Lysandre Debut committed
16
    "diffusers": "diffusers",
17
    "dill": "dill<0.3.5",
18
    "evaluate": "evaluate>=0.2.0",
19
20
21
    "faiss-cpu": "faiss-cpu",
    "fastapi": "fastapi",
    "filelock": "filelock",
22
    "flax": "flax>=0.4.1,<=0.7.0",
23
    "fsspec": "fsspec<2023.10.0",
24
    "ftfy": "ftfy",
25
    "fugashi": "fugashi>=1.0",
26
    "GitPython": "GitPython<3.1.19",
27
    "hf-doc-builder": "hf-doc-builder>=0.3.0",
28
    "huggingface-hub": "huggingface-hub>=0.16.4,<1.0",
29
    "importlib_metadata": "importlib_metadata",
30
31
    "ipadic": "ipadic>=1.0.0,<2.0",
    "isort": "isort>=5.5.4",
32
33
    "jax": "jax>=0.4.1,<=0.4.13",
    "jaxlib": "jaxlib>=0.4.1,<=0.4.13",
34
    "jieba": "jieba",
35
    "kenlm": "kenlm",
Matt's avatar
Matt committed
36
    "keras": "keras<2.16",
37
    "keras-nlp": "keras-nlp>=0.3.1",
Sylvain Gugger's avatar
Sylvain Gugger committed
38
    "librosa": "librosa",
39
    "nltk": "nltk",
40
    "natten": "natten>=0.14.6",
Sylvain Gugger's avatar
Sylvain Gugger committed
41
    "numpy": "numpy>=1.17",
42
43
44
    "onnxconverter-common": "onnxconverter-common",
    "onnxruntime-tools": "onnxruntime-tools>=1.4.2",
    "onnxruntime": "onnxruntime>=1.4.0",
Lysandre Debut's avatar
Lysandre Debut committed
45
    "opencv-python": "opencv-python",
Lysandre Debut's avatar
Lysandre Debut committed
46
    "optuna": "optuna",
47
    "optax": "optax>=0.0.8,<=0.1.4",
48
    "packaging": "packaging>=20.0",
49
    "parameterized": "parameterized",
50
    "phonemizer": "phonemizer",
Yih-Dar's avatar
Yih-Dar committed
51
    "protobuf": "protobuf",
52
    "psutil": "psutil",
53
    "pyyaml": "pyyaml>=5.1",
54
    "pydantic": "pydantic<2",
55
    "pytest": "pytest>=7.2.0",
Lysandre Debut's avatar
Lysandre Debut committed
56
    "pytest-timeout": "pytest-timeout",
57
    "pytest-xdist": "pytest-xdist",
58
    "python": "python>=3.8.0",
Sylvain Gugger's avatar
Style  
Sylvain Gugger committed
59
    "ray[tune]": "ray[tune]",
60
61
    "regex": "regex!=2019.12.17",
    "requests": "requests",
Sylvain Gugger's avatar
Sylvain Gugger committed
62
    "rhoknp": "rhoknp>=1.1.0,<1.3.1",
63
    "rjieba": "rjieba",
64
    "rouge-score": "rouge-score!=0.0.7,!=0.0.8,!=0.1,!=0.1.1",
Sylvain Gugger's avatar
Sylvain Gugger committed
65
    "ruff": "ruff>=0.0.241,<=0.0.259",
Sylvain Gugger's avatar
Sylvain Gugger committed
66
    "sacrebleu": "sacrebleu>=1.4.12,<2.0.0",
67
    "sacremoses": "sacremoses",
68
    "safetensors": "safetensors>=0.3.1",
69
    "sagemaker": "sagemaker>=2.31.0",
70
    "scikit-learn": "scikit-learn",
71
    "sentencepiece": "sentencepiece>=0.1.91,!=0.1.92",
72
    "sigopt": "sigopt",
73
    "starlette": "starlette",
Sylvain Gugger's avatar
Sylvain Gugger committed
74
75
    "sudachipy": "sudachipy>=0.6.6",
    "sudachidict_core": "sudachidict_core>=20220729",
76
    "tensorboard": "tensorboard",
Matt's avatar
Matt committed
77
78
79
    "tensorflow-cpu": "tensorflow-cpu>=2.6,<2.16",
    "tensorflow": "tensorflow>=2.6,<2.16",
    "tensorflow-text": "tensorflow-text<2.16",
80
    "tf2onnx": "tf2onnx",
81
    "timeout-decorator": "timeout-decorator",
NielsRogge's avatar
NielsRogge committed
82
    "timm": "timm",
83
    "tokenizers": "tokenizers>=0.14,<0.19",
Yih-Dar's avatar
Yih-Dar committed
84
    "torch": "torch>=1.10,!=1.12.0",
Suraj Patil's avatar
Suraj Patil committed
85
    "torchaudio": "torchaudio",
NielsRogge's avatar
NielsRogge committed
86
    "torchvision": "torchvision",
87
    "pyctcdecode": "pyctcdecode>=0.4.0",
88
89
90
    "tqdm": "tqdm>=4.27",
    "unidic": "unidic>=1.0.2",
    "unidic_lite": "unidic_lite>=1.0.7",
Sylvain Gugger's avatar
Sylvain Gugger committed
91
    "urllib3": "urllib3<2.0.0",
92
93
    "uvicorn": "uvicorn",
}