pyrox.dev
1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5
6 # dependencies
7 einops,
8 numpy,
9 torch,
10 torchaudio,
11}:
12
13buildPythonPackage rec {
14 pname = "encodec";
15 version = "0.1.1";
16 format = "setuptools";
17
18 src = fetchFromGitHub {
19 owner = "facebookresearch";
20 repo = "encodec";
21 rev = "v${version}";
22 hash = "sha256-+iJZkX1HoyuNFu9VRxMO6aAzNQybkH9lrQJ5Ao9+/CY=";
23 };
24
25 propagatedBuildInputs = [
26 einops
27 numpy
28 torch
29 torchaudio
30 ];
31
32 pythonImportsCheck = [ "encodec" ];
33
34 # requires model data from the internet
35 doCheck = false;
36
37 meta = with lib; {
38 description = "State-of-the-art deep learning based audio codec supporting both mono 24 kHz audio and stereo 48 kHz audio";
39 homepage = "https://github.com/facebookresearch/encodec";
40 changelog = "https://github.com/facebookresearch/encodec/blob/${src.rev}/CHANGELOG.md";
41 license = licenses.mit;
42 teams = [ teams.tts ];
43 };
44}