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