diff --git a/482_WhisperX/WhisperX.eb b/482_WhisperX/WhisperX.eb index f1239c3..1487b14 100644 --- a/482_WhisperX/WhisperX.eb +++ b/482_WhisperX/WhisperX.eb @@ -12,17 +12,32 @@ toolchain = {'name': 'foss', 'version': '2023a'} # DEPS: # OK torch>=2 # OK torchaudio>=2 - # OK? faster-whisper==1.0.0 + # OK faster-whisper==1.0.0 # OK CREATED ctranslate2==4.4.0 # OK transformers # OK pandas # OK setuptools>=65 # OK nltk + # ok pyannote.audio==3.1.1 # EXTS: - # !!! whisperx 3.2.0 has requirement torchaudio>=2, but you have torchaudio 0.16.2. - - + # !!! whisperx 3.2.0 has requirement torchaudio>=2, but you have torchaudio 0.16.2. -> E1 + # pyannote-audio 3.1.1 requires asteroid-filterbanks, which is not installed. + # pyannote-audio 3.1.1 requires einops, which is not installed. + # pyannote-audio 3.1.1 requires lightning, which is not installed. + # pyannote-audio 3.1.1 requires omegaconf, which is not installed. + # pyannote-audio 3.1.1 requires pyannote-core, which is not installed. + # pyannote-audio 3.1.1 requires pyannote-database, which is not installed. + # pyannote-audio 3.1.1 requires pyannote-metrics, which is not installed. + # pyannote-audio 3.1.1 requires pyannote-pipeline, which is not installed. + # pyannote-audio 3.1.1 requires pytorch-metric-learning, which is not installed. + # pyannote-audio 3.1.1 requires semver, which is not installed. + # pyannote-audio 3.1.1 requires soundfile, which is not installed. + # pyannote-audio 3.1.1 requires speechbrain, which is not installed. + # pyannote-audio 3.1.1 requires tensorboardx, which is not installed. + # pyannote-audio 3.1.1 requires torch-audiomentations, which is not installed. + # pyannote-audio 3.1.1 requires torchmetrics, which is not installed. + # pyannote-audio 3.1.1 has requirement torchaudio>=2.0.0, but you have torchaudio 0.16.2. dependencies = [ ('CUDA', '12.1.1', '', SYSTEM), @@ -35,6 +50,7 @@ dependencies = [ ('FFmpeg', '6.0'), ('PyAV', '11.0.0'), ('ONNX-Runtime', '1.19.2', versionsuffix), + # pyannote-bundle: https://github.com/easybuilders/easybuild-easyconfigs/pull/19992/ ] use_pip = True diff --git a/482_WhisperX/pyannote-bundle.eb b/482_WhisperX/pyannote-bundle.eb new file mode 100644 index 0000000..950fe3b --- /dev/null +++ b/482_WhisperX/pyannote-bundle.eb @@ -0,0 +1,61 @@ +easyblock = 'PythonBundle' + +name = 'pyannote' +version = '2025.01' +local_pytorch_version = '2.1.2' +local_cuda_version = '-CUDA-%(cudaver)s' +versionsuffix = '-PyTorch-' + local_pytorch_version + local_cuda_version + + +homepage = 'https://github.com/pyannote' +description = "pyannote is an open-source toolkit for speaker diarization." + +toolchain = {'name': 'foss', 'version': '2023a'} + +dependencies = [ + ('CUDA', '12.1.1', '', SYSTEM), + ('Python', '3.11.3'), + ('PyTorch-bundle', '2.1.2', local_cuda_version), + # ('PyTorch-Lightning', '2.2.0.post0', '-CUDA-%(cudaver)s'), + # ('SciPy-bundle', '2023.07'), + # ('scikit-learn', '1.3.1'), + # ('PyYAML', '6.0'), + # ('Optuna', '3.5.0'), + # ('matplotlib', '3.7.2'), + # ('sympy', '1.12'), + # ('typer', '0.9.0'), + # ('einops', '0.7.0'), + # ('torchaudio', '2.2.0', versionsuffix), + # ('torch-audiomentations', '0.11.1', versionsuffix), + # ('huggingface-hub', '0.20.3'), + # ('asteroid-filterbanks', '0.4.0', versionsuffix), + # ('speechbrain', '0.5.16', versionsuffix), + # ('pytorch-metric-learning', '2.4.1', versionsuffix), + # ('semver', '3.0.2'), + # ('omegaconf', '2.3.0', '-Java-11'), + # ('lightning', '2.2.0.post0', versionsuffix), +] + +use_pip = True + +exts_list = [ + # ('pyannote.core', '5.0.0', { + # 'checksums': ['1a55bcc8bd680ba6be5fa53efa3b6f3d2cdd67144c07b6b4d8d66d5cb0d2096f'], + # }), + # ('pyannote.database', '5.0.1', { + # 'checksums': ['980bdc910615380a17d8bd82805f9770006d050b2c7a946a520ade4627192681'], + # }), + # ('pyannote.metrics', '3.2.1', { + # 'checksums': ['08024255a3550e96a8e9da4f5f4af326886548480de891414567c8900920ee5c'], + # }), + # ('pyannote.pipeline', '3.0.1', { + # 'checksums': ['021794e26a2cf5d8fb5bb1835951e71f5fac33eb14e23dfb7468e16b1b805151'], + # }), + ('pyannote.audio', '3.1.1', { + 'checksums': ['b6562d46b5d5d616c1887cafd5aab3f7d19ecf6b3312fc6c4f72f37332fcd408'], + }), +] + +sanity_pip_check = True + +moduleclass = 'tools' \ No newline at end of file