tjh.dev
1{
2 buildPythonPackage,
3 fetchFromGitHub,
4 huggingface-hub,
5 hyperpyyaml,
6 joblib,
7 lib,
8 numpy,
9 packaging,
10 pythonOlder,
11 sentencepiece,
12 scipy,
13 torch,
14 torchaudio,
15 tqdm,
16}:
17
18buildPythonPackage rec {
19 pname = "speechbrain";
20 version = "1.0.0";
21 format = "setuptools";
22
23 disabled = pythonOlder "3.7";
24
25 src = fetchFromGitHub {
26 owner = "speechbrain";
27 repo = "speechbrain";
28 rev = "refs/tags/v${version}";
29 hash = "sha256-s23RHxqr+ZCWPYNchFpyYa01mY+Bum3thgNbJP8rAmA=";
30 };
31
32 propagatedBuildInputs = [
33 huggingface-hub
34 hyperpyyaml
35 joblib
36 numpy
37 packaging
38 sentencepiece
39 scipy
40 torch
41 torchaudio
42 tqdm
43 ];
44
45 doCheck = false; # requires sox backend
46
47 pythonImportsCheck = [ "speechbrain" ];
48
49 meta = with lib; {
50 description = "PyTorch-based Speech Toolkit";
51 homepage = "https://speechbrain.github.io";
52 changelog = "https://github.com/speechbrain/speechbrain/releases/tag/v${version}";
53 license = licenses.asl20;
54 maintainers = with maintainers; [ GaetanLepage ];
55 };
56}