Skip to content

Commit

Permalink
Merge pull request #22031 from boegel/20241215173242_new_pr_Whisper20…
Browse files Browse the repository at this point in the history
…240930

{ai}[foss/2023a] Whisper v20240930 w/ CUDA 12.1.1
  • Loading branch information
laraPPr authored Dec 19, 2024
2 parents 45813a2 + 9a8105f commit c31c9c7
Showing 1 changed file with 41 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
easyblock = 'PythonBundle'

name = 'Whisper'
version = '20240930'
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),
('FFmpeg', '6.0'),
]

use_pip = True

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

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 c31c9c7

Please sign in to comment.