diff --git a/pkgs/development/python-modules/curated-transformers/default.nix b/pkgs/development/python-modules/curated-transformers/default.nix new file mode 100644 index 000000000000..32a52a5fd06a --- /dev/null +++ b/pkgs/development/python-modules/curated-transformers/default.nix @@ -0,0 +1,36 @@ +{ + lib, + buildPythonPackage, + fetchFromGitHub, + setuptools, + torch, +}: + +buildPythonPackage rec { + pname = "curated-transformers"; + version = "0.1.1"; + pyproject = true; + + src = fetchFromGitHub { + owner = "explosion"; + repo = "curated-transformers"; + tag = "v${version}"; + hash = "sha256-QhJZnQIa9TilwdQCUlxnQCEc6Suj669cht6WHUAr/Gw="; + }; + + build-system = [ setuptools ]; + + dependencies = [ torch ]; + + # Unit tests are hard to use, since most tests rely on downloading + # models from Hugging Face Hub. + pythonImportCheck = [ "curated_transformers" ]; + + meta = with lib; { + description = "PyTorch library of curated Transformer models and their composable components"; + homepage = "https://github.com/explosion/curated-transformers"; + changelog = "https://github.com/explosion/curated-transformers/releases/tag/v${version}"; + license = licenses.mit; + maintainers = with maintainers; [ danieldk ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 585511d44bf1..6af7b60b7a02 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -2793,6 +2793,8 @@ self: super: with self; { curated-tokenizers = callPackage ../development/python-modules/curated-tokenizers { }; + curated-transformers = callPackage ../development/python-modules/curated-transformers { }; + customtkinter = callPackage ../development/python-modules/customtkinter { }; cucumber-tag-expressions = callPackage ../development/python-modules/cucumber-tag-expressions { };