requirements.txt

#
# This file is autogenerated by pip-compile with Python 3.10
# by the following command:
#
#    pip-compile
#
absl-py==2.1.0
    # via
    #   tensorboard
    #   tensorflow
aiohttp==3.9.5
    # via fsspec
aiosignal==1.3.1
    # via aiohttp
alembic==1.13.2
    # via optuna
antlr4-python3-runtime==4.9.3
    # via omegaconf
appdirs==1.4.4
    # via pyhyphen-reloaded
asteroid-filterbanks==0.4.0
    # via pyannote-audio
astroid==3.2.4
    # via pylint
astunparse==1.6.3
    # via tensorflow
async-timeout==4.0.3
    # via aiohttp
attrs==23.2.0
    # via aiohttp
audioread==3.0.1
    # via librosa
av==11.0.0
    # via faster-whisper
black==24.4.2
    # via -r requirements.in
brotli==1.1.0
    # via yt-dlp
cachetools==5.4.0
    # via google-auth
certifi==2024.7.4
    # via
    #   requests
    #   yt-dlp
cffi==1.16.0
    # via soundfile
chardet==5.2.0
    # via music21
charset-normalizer==3.3.2
    # via requests
click==8.1.7
    # via
    #   black
    #   nltk
    #   typer
cloudpickle==3.0.0
    # via submitit
colorama==0.4.6
    # via
    #   click
    #   colorlog
    #   pylint
    #   pytest
    #   tqdm
coloredlogs==15.0.1
    # via onnxruntime
colorlog==6.8.2
    # via optuna
contourpy==1.2.1
    # via matplotlib
crepe==0.0.15
    # via -r requirements.in
ctranslate2==4.3.1
    # via faster-whisper
cycler==0.12.1
    # via matplotlib
dataclasses==0.6
    # via -r requirements.in
dataclasses-json==0.6.7
    # via -r requirements.in
decorator==5.1.1
    # via librosa
demucs==4.0.1
    # via -r requirements.in
dill==0.3.8
    # via pylint
docopt==0.6.2
    # via
    #   pyannote-metrics
    #   pyannote-pipeline
dora-search==0.1.12
    # via demucs
einops==0.8.0
    # via
    #   demucs
    #   pyannote-audio
exceptiongroup==1.2.2
    # via pytest
faster-whisper==1.0.0
    # via whisperx
ffmpeg-python==0.2.0
    # via -r requirements.in
filelock==3.15.4
    # via
    #   huggingface-hub
    #   pyannote-pipeline
    #   torch
    #   transformers
flatbuffers==24.3.25
    # via
    #   onnxruntime
    #   tensorflow
fonttools==4.53.1
    # via matplotlib
frozenlist==1.4.1
    # via
    #   aiohttp
    #   aiosignal
fsspec[http]==2024.6.1
    # via
    #   huggingface-hub
    #   lightning
    #   pytorch-lightning
    #   torch
future==1.0.0
    # via ffmpeg-python
gast==0.4.0
    # via tensorflow
google-auth==2.32.0
    # via
    #   google-auth-oauthlib
    #   tensorboard
google-auth-oauthlib==0.4.6
    # via tensorboard
google-pasta==0.2.0
    # via tensorflow
greenlet==3.0.3
    # via sqlalchemy
grpcio==1.65.1
    # via
    #   tensorboard
    #   tensorflow
h5py==3.11.0
    # via
    #   crepe
    #   tensorflow
hmmlearn==0.3.2
    # via crepe
huggingface-hub==0.24.2
    # via
    #   faster-whisper
    #   pyannote-audio
    #   speechbrain
    #   tokenizers
    #   transformers
humanfriendly==10.0
    # via coloredlogs
hyperpyyaml==1.2.2
    # via speechbrain
idna==3.7
    # via
    #   requests
    #   yarl
imageio==2.34.2
    # via crepe
iniconfig==2.0.0
    # via pytest
isort==5.13.2
    # via
    #   -r requirements.in
    #   pylint
jinja2==3.1.4
    # via torch
joblib==1.4.2
    # via
    #   librosa
    #   music21
    #   nltk
    #   scikit-learn
    #   speechbrain
jsonpickle==3.2.2
    # via music21
julius==0.2.7
    # via
    #   demucs
    #   torch-audiomentations
keras==2.10.0
    # via tensorflow
keras-preprocessing==1.1.2
    # via tensorflow
kiwisolver==1.4.5
    # via matplotlib
lameenc==1.7.0
    # via demucs
langcodes==3.4.0
    # via -r requirements.in
language-data==1.2.0
    # via
    #   -r requirements.in
    #   langcodes
levenshtein==0.25.1
    # via python-levenshtein
libclang==18.1.1
    # via tensorflow
librosa==0.9.2
    # via
    #   -r requirements.in
    #   torch-audiomentations
lightning==2.3.3
    # via pyannote-audio
lightning-utilities==0.11.6
    # via
    #   lightning
    #   pytorch-lightning
    #   torchmetrics
llvmlite==0.43.0
    # via numba
mako==1.3.5
    # via alembic
marisa-trie==1.2.0
    # via language-data
markdown==3.6
    # via tensorboard
markdown-it-py==3.0.0
    # via rich
markupsafe==2.1.5
    # via
    #   jinja2
    #   mako
    #   werkzeug
marshmallow==3.21.3
    # via dataclasses-json
matplotlib==3.9.2
    # via
    #   -r requirements.in
    #   crepe
    #   music21
    #   pyannote-metrics
mccabe==0.7.0
    # via pylint
mdurl==0.1.2
    # via markdown-it-py
mido==1.3.2
    # via pretty-midi
more-itertools==10.3.0
    # via music21
mpmath==1.3.0
    # via sympy
multidict==6.0.5
    # via
    #   aiohttp
    #   yarl
music21==9.1.0
    # via -r requirements.in
musicbrainzngs==0.7.1
    # via -r requirements.in
mutagen==1.47.0
    # via yt-dlp
mypy-extensions==1.0.0
    # via
    #   black
    #   typing-inspect
networkx==3.3
    # via torch
nltk==3.8.1
    # via whisperx
numba==0.60.0
    # via
    #   librosa
    #   resampy
numpy==1.26.4
    # via
    #   -r requirements.in
    #   asteroid-filterbanks
    #   contourpy
    #   crepe
    #   ctranslate2
    #   h5py
    #   hmmlearn
    #   imageio
    #   keras-preprocessing
    #   librosa
    #   lightning
    #   matplotlib
    #   music21
    #   numba
    #   onnxruntime
    #   openunmix
    #   opt-einsum
    #   optuna
    #   pandas
    #   pretty-midi
    #   pyannote-core
    #   pyannote-metrics
    #   pytorch-lightning
    #   pytorch-metric-learning
    #   resampy
    #   scikit-learn
    #   scipy
    #   speechbrain
    #   tensorboard
    #   tensorboardx
    #   tensorflow
    #   torchmetrics
    #   transformers
oauthlib==3.2.2
    # via requests-oauthlib
omegaconf==2.3.0
    # via
    #   dora-search
    #   pyannote-audio
onnxruntime==1.18.1
    # via faster-whisper
openunmix==1.3.0
    # via demucs
opt-einsum==3.3.0
    # via tensorflow
optuna==3.6.1
    # via pyannote-pipeline
packaging==23.2
    # via
    #   -r requirements.in
    #   black
    #   huggingface-hub
    #   librosa
    #   lightning
    #   lightning-utilities
    #   marshmallow
    #   matplotlib
    #   mido
    #   onnxruntime
    #   optuna
    #   pooch
    #   pytest
    #   pytorch-lightning
    #   speechbrain
    #   tensorboardx
    #   tensorflow
    #   torch-pitch-shift
    #   torchmetrics
    #   transformers
pandas==2.2.2
    # via
    #   pyannote-database
    #   pyannote-metrics
    #   whisperx
pathspec==0.12.1
    # via black
pillow==10.4.0
    # via
    #   -r requirements.in
    #   imageio
    #   matplotlib
platformdirs==4.2.2
    # via
    #   black
    #   pooch
    #   pylint
pluggy==1.5.0
    # via pytest
pooch==1.8.2
    # via librosa
pretty-midi==0.2.10
    # via -r requirements.in
primepy==1.3
    # via torch-pitch-shift
protobuf==3.19.6
    # via
    #   -r requirements.in
    #   onnxruntime
    #   tensorboard
    #   tensorboardx
    #   tensorflow
pyannote-audio==3.1.1
    # via whisperx
pyannote-core==5.0.0
    # via
    #   pyannote-audio
    #   pyannote-database
    #   pyannote-metrics
    #   pyannote-pipeline
pyannote-database==5.1.0
    # via
    #   pyannote-audio
    #   pyannote-metrics
    #   pyannote-pipeline
pyannote-metrics==3.2.1
    # via pyannote-audio
pyannote-pipeline==3.0.1
    # via pyannote-audio
pyasn1==0.6.0
    # via
    #   pyasn1-modules
    #   rsa
pyasn1-modules==0.4.0
    # via google-auth
pycparser==2.22
    # via cffi
pycryptodomex==3.20.0
    # via yt-dlp
pydub==0.25.1
    # via -r requirements.in
pygments==2.18.0
    # via rich
pyhyphen-reloaded==4.0.6
    # via -r requirements.in
pylint==3.2.6
    # via -r requirements.in
pyparsing==3.1.2
    # via matplotlib
pyreadline3==3.4.1
    # via humanfriendly
pytest==8.3.2
    # via -r requirements.in
python-dateutil==2.9.0.post0
    # via
    #   matplotlib
    #   pandas
python-levenshtein==0.25.1
    # via -r requirements.in
pytorch-lightning==2.3.3
    # via lightning
pytorch-metric-learning==2.6.0
    # via pyannote-audio
pytz==2024.1
    # via pandas
pyyaml==6.0.1
    # via
    #   ctranslate2
    #   demucs
    #   huggingface-hub
    #   hyperpyyaml
    #   lightning
    #   omegaconf
    #   optuna
    #   pyannote-database
    #   pyannote-pipeline
    #   pytorch-lightning
    #   transformers
rapidfuzz==3.9.4
    # via levenshtein
regex==2024.7.24
    # via
    #   nltk
    #   transformers
requests==2.32.3
    # via
    #   huggingface-hub
    #   music21
    #   pooch
    #   pyhyphen-reloaded
    #   requests-oauthlib
    #   tensorboard
    #   transformers
    #   yt-dlp
requests-oauthlib==2.0.0
    # via google-auth-oauthlib
resampy==0.2.2
    # via
    #   crepe
    #   librosa
retrying==1.3.4
    # via dora-search
rich==13.7.1
    # via
    #   pyannote-audio
    #   typer
rsa==4.9
    # via google-auth
ruamel-yaml==0.18.6
    # via hyperpyyaml
ruamel-yaml-clib==0.2.8
    # via ruamel-yaml
safetensors==0.4.3
    # via transformers
scikit-learn==1.5.1
    # via
    #   crepe
    #   hmmlearn
    #   librosa
    #   pyannote-metrics
    #   pyannote-pipeline
    #   pytorch-metric-learning
scipy==1.14.0
    # via
    #   -r requirements.in
    #   crepe
    #   hmmlearn
    #   librosa
    #   pyannote-core
    #   pyannote-metrics
    #   resampy
    #   scikit-learn
    #   speechbrain
semver==3.0.2
    # via pyannote-audio
sentencepiece==0.2.0
    # via speechbrain
shellingham==1.5.4
    # via typer
six==1.16.0
    # via
    #   astunparse
    #   google-pasta
    #   keras-preprocessing
    #   pretty-midi
    #   python-dateutil
    #   resampy
    #   retrying
    #   tensorflow
sortedcontainers==2.4.0
    # via pyannote-core
soundfile==0.12.1
    # via
    #   librosa
    #   pyannote-audio
speechbrain==1.0.0
    # via pyannote-audio
sqlalchemy==2.0.31
    # via
    #   alembic
    #   optuna
submitit==1.5.1
    # via dora-search
sympy==1.13.1
    # via
    #   onnxruntime
    #   pyannote-metrics
    #   torch
tabulate==0.9.0
    # via pyannote-metrics
tensorboard==2.10.1
    # via tensorflow
tensorboard-data-server==0.6.1
    # via tensorboard
tensorboard-plugin-wit==1.8.1
    # via tensorboard
tensorboardx==2.6
    # via pyannote-audio
tensorflow==2.10.0
    # via -r requirements.in
tensorflow-estimator==2.10.0
    # via tensorflow
tensorflow-io-gcs-filesystem==0.31.0
    # via tensorflow
termcolor==2.4.0
    # via tensorflow
threadpoolctl==3.5.0
    # via scikit-learn
tokenizers==0.15.2
    # via
    #   faster-whisper
    #   transformers
tomli==2.0.1
    # via
    #   black
    #   pylint
    #   pytest
tomlkit==0.13.0
    # via pylint
torch==2.4.0
    # via
    #   asteroid-filterbanks
    #   demucs
    #   dora-search
    #   julius
    #   lightning
    #   openunmix
    #   pyannote-audio
    #   pytorch-lightning
    #   pytorch-metric-learning
    #   speechbrain
    #   torch-audiomentations
    #   torch-pitch-shift
    #   torchaudio
    #   torchmetrics
    #   whisperx
torch-audiomentations==0.11.1
    # via pyannote-audio
torch-pitch-shift==1.2.4
    # via torch-audiomentations
torchaudio==2.4.0
    # via
    #   demucs
    #   openunmix
    #   pyannote-audio
    #   speechbrain
    #   torch-audiomentations
    #   torch-pitch-shift
    #   whisperx
torchmetrics==1.4.0.post0
    # via
    #   lightning
    #   pyannote-audio
    #   pytorch-lightning
tqdm==4.66.4
    # via
    #   -r requirements.in
    #   demucs
    #   huggingface-hub
    #   lightning
    #   nltk
    #   openunmix
    #   optuna
    #   pyannote-pipeline
    #   pytorch-lightning
    #   pytorch-metric-learning
    #   speechbrain
    #   transformers
transformers==4.39.3
    # via whisperx
treetable==0.2.5
    # via dora-search
typer==0.12.3
    # via pyannote-database
typing-extensions==4.12.2
    # via
    #   alembic
    #   asteroid-filterbanks
    #   astroid
    #   black
    #   huggingface-hub
    #   lightning
    #   lightning-utilities
    #   pyannote-core
    #   pytorch-lightning
    #   sqlalchemy
    #   submitit
    #   tensorflow
    #   torch
    #   typer
    #   typing-inspect
typing-inspect==0.9.0
    # via dataclasses-json
tzdata==2024.1
    # via pandas
unidecode==1.3.8
    # via -r requirements.in
urllib3==2.2.2
    # via
    #   requests
    #   yt-dlp
webcolors==24.6.0
    # via music21
websockets==12.0
    # via yt-dlp
werkzeug==3.0.3
    # via tensorboard
wheel==0.43.0
    # via
    #   astunparse
    #   pyhyphen-reloaded
    #   tensorboard
whisperx @ git+https://github.com/m-bain/whisperx.git
    # via -r requirements.in
wrapt==1.16.0
    # via tensorflow
yarl==1.9.4
    # via aiohttp
yt-dlp==2024.7.25
    # via -r requirements.in

# The following packages are considered to be unsafe in a requirements file:
# setuptools