pkgs/development/python-modules/encodec/default.nix

   1 {
   2   lib,
   3   buildPythonPackage,
   4   fetchFromGitHub,
   5
   6   # dependencies
   7   einops,
   8   numpy,
   9   torch-bin,
  10   torchaudio-bin,
  11 }:
  12
  13 buildPythonPackage 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-bin
  29     torchaudio-bin
  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     maintainers = teams.tts.members;
  43   };
  44 }