Skip to content

Commit

Permalink
Merge pull request #21470 from boegel/20240923154007_new_pr_Whisper20…
Browse files Browse the repository at this point in the history
…231117

{ai}[foss/2023a] Whisper v20231117 w/ CUDA 12.1.1
  • Loading branch information
laraPPr authored Sep 25, 2024
2 parents 50fa48d + a6c88f9 commit 3f28f23
Showing 1 changed file with 40 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,40 @@
easyblock = 'PythonBundle'

name = 'Whisper'
version = '20231117'
versionsuffix = '-CUDA-12.1.1'

homepage = 'https://github.com/openai/whisper'
description = "Whisper is a general-purpose speech recognition model"

toolchain = {'name': 'foss', 'version': '2023a'}

dependencies = [
('Python', '3.11.3'),
('SciPy-bundle', '2023.07'),
('tqdm', '4.66.1'),
('numba', '0.58.1'),
('PyTorch', '2.1.2', versionsuffix),
('tiktoken', '0.7.0'),
('Triton', '2.1.0', versionsuffix),
]

use_pip = True

exts_list = [
('openai-whisper', version, {
'checksums': ['7af424181436f1800cc0b7d75cf40ede34e9ddf1ba4983a910832fcf4aade4a4'],
'modulename': 'whisper',
}),
]

sanity_check_paths = {
'files': ['bin/whisper'],
'dirs': ['lib/python%(pyshortver)s/site-packages'],
}

sanity_check_commands = ["whisper --help"]

sanity_pip_check = True

moduleclass = 'ai'

0 comments on commit 3f28f23

Please sign in to comment.