| | |
| | |
| | |
| | deps = { |
| | "Pillow": "Pillow", |
| | "accelerate": "accelerate>=0.31.0", |
| | "compel": "compel==0.1.8", |
| | "datasets": "datasets", |
| | "filelock": "filelock", |
| | "flax": "flax>=0.4.1", |
| | "hf-doc-builder": "hf-doc-builder>=0.3.0", |
| | "huggingface-hub": "huggingface-hub>=0.23.2", |
| | "requests-mock": "requests-mock==1.10.0", |
| | "importlib_metadata": "importlib_metadata", |
| | "invisible-watermark": "invisible-watermark>=0.2.0", |
| | "isort": "isort>=5.5.4", |
| | "jax": "jax>=0.4.1", |
| | "jaxlib": "jaxlib>=0.4.1", |
| | "Jinja2": "Jinja2", |
| | "k-diffusion": "k-diffusion>=0.0.12", |
| | "torchsde": "torchsde", |
| | "note_seq": "note_seq", |
| | "librosa": "librosa", |
| | "numpy": "numpy", |
| | "parameterized": "parameterized", |
| | "peft": "peft>=0.6.0", |
| | "protobuf": "protobuf>=3.20.3,<4", |
| | "pytest": "pytest", |
| | "pytest-timeout": "pytest-timeout", |
| | "pytest-xdist": "pytest-xdist", |
| | "python": "python>=3.8.0", |
| | "ruff": "ruff==0.1.5", |
| | "safetensors": "safetensors>=0.3.1", |
| | "sentencepiece": "sentencepiece>=0.1.91,!=0.1.92", |
| | "GitPython": "GitPython<3.1.19", |
| | "scipy": "scipy", |
| | "onnx": "onnx", |
| | "regex": "regex!=2019.12.17", |
| | "requests": "requests", |
| | "tensorboard": "tensorboard", |
| | "torch": "torch>=1.4", |
| | "torchvision": "torchvision", |
| | "transformers": "transformers>=4.41.2", |
| | "urllib3": "urllib3<=2.0.0", |
| | "black": "black", |
| | } |
| |
|