| |
| |
| |
| 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", |
| } |
|
|