Skip to content

Commit

Permalink
adding easyconfigs: Whisper-20231117-foss-2023a-CUDA-12.1.1.eb
Browse files Browse the repository at this point in the history
  • Loading branch information
boegel committed Sep 23, 2024
1 parent fedb068 commit a6c88f9
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 a6c88f9

Please sign in to comment.