diff options
Diffstat (limited to 'pkgs/development/python-modules/rotary-embedding-torch/default.nix')
-rw-r--r-- | pkgs/development/python-modules/rotary-embedding-torch/default.nix | 48 |
1 files changed, 48 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/rotary-embedding-torch/default.nix b/pkgs/development/python-modules/rotary-embedding-torch/default.nix new file mode 100644 index 00000000000..8a65e0249e7 --- /dev/null +++ b/pkgs/development/python-modules/rotary-embedding-torch/default.nix @@ -0,0 +1,48 @@ +{ lib +, buildPythonPackage +, fetchFromGitHub + +# build-system +, setuptools +, wheel + +# dependencies +, einops +, torch +}: + +buildPythonPackage rec { + pname = "rotary-embedding-torch"; + version = "0.3.5"; + pyproject = true; + + src = fetchFromGitHub { + owner = "lucidrains"; + repo = "rotary-embedding-torch"; + rev = version; + hash = "sha256-dST3eJnOcG2s9tiD/Fb9BvLS6nIpE8RXly92PK/gCC8="; + }; + + nativeBuildInputs = [ + setuptools + wheel + ]; + + propagatedBuildInputs = [ + einops + torch + ]; + + pythonImportsCheck = [ + "rotary_embedding_torch" + ]; + + doCheck = false; # no tests + + meta = with lib; { + description = "Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch"; + homepage = "https://github.com/lucidrains/rotary-embedding-torch"; + license = licenses.mit; + maintainers = teams.tts.members; + }; +} |