dependency_versions_table.py 1.45 KB
Newer Older
1
2
# THIS FILE HAS BEEN AUTOGENERATED. To update:
# 1. modify the `_deps` dict in setup.py
3
# 2. run `make deps_table_update`
4
deps = {
5
    "Pillow": "Pillow",
Anton Lozhkov's avatar
Anton Lozhkov committed
6
    "accelerate": "accelerate>=0.11.0",
7
    "compel": "compel==0.1.8",
8
    "datasets": "datasets",
Patrick von Platen's avatar
Patrick von Platen committed
9
    "filelock": "filelock",
10
    "flax": "flax>=0.4.1",
Anton Lozhkov's avatar
Anton Lozhkov committed
11
    "hf-doc-builder": "hf-doc-builder>=0.3.0",
Lucain's avatar
Lucain committed
12
    "huggingface-hub": "huggingface-hub>=0.20.2",
13
    "requests-mock": "requests-mock==1.10.0",
Patrick von Platen's avatar
Patrick von Platen committed
14
    "importlib_metadata": "importlib_metadata",
15
    "invisible-watermark": "invisible-watermark>=0.2.0",
16
    "isort": "isort>=5.5.4",
17
18
    "jax": "jax>=0.4.1",
    "jaxlib": "jaxlib>=0.4.1",
Lucain's avatar
Lucain committed
19
    "Jinja2": "Jinja2",
20
    "k-diffusion": "k-diffusion>=0.0.12",
21
22
    "torchsde": "torchsde",
    "note_seq": "note_seq",
23
    "librosa": "librosa",
Patrick von Platen's avatar
Patrick von Platen committed
24
    "numpy": "numpy",
25
    "parameterized": "parameterized",
Patrick von Platen's avatar
Patrick von Platen committed
26
    "peft": "peft>=0.6.0",
27
    "protobuf": "protobuf>=3.20.3,<4",
28
    "pytest": "pytest",
29
30
    "pytest-timeout": "pytest-timeout",
    "pytest-xdist": "pytest-xdist",
31
    "python": "python>=3.8.0",
Dhruv Nair's avatar
Dhruv Nair committed
32
    "ruff": "ruff==0.1.5",
33
    "safetensors": "safetensors>=0.3.1",
Suraj Patil's avatar
Suraj Patil committed
34
    "sentencepiece": "sentencepiece>=0.1.91,!=0.1.92",
Dhruv Nair's avatar
Dhruv Nair committed
35
    "GitPython": "GitPython<3.1.19",
36
    "scipy": "scipy",
37
    "onnx": "onnx",
38
39
    "regex": "regex!=2019.12.17",
    "requests": "requests",
40
    "tensorboard": "tensorboard",
41
42
    "torch": "torch>=1.4",
    "torchvision": "torchvision",
43
    "transformers": "transformers>=4.25.1",
44
    "urllib3": "urllib3<=2.0.0",
45
}