diff --git a/pkgs/development/python-modules/rotary-embedding-torch/default.nix b/pkgs/development/python-modules/rotary-embedding-torch/default.nix index 91bb5a00231c..f9467f63e48c 100644 --- a/pkgs/development/python-modules/rotary-embedding-torch/default.nix +++ b/pkgs/development/python-modules/rotary-embedding-torch/default.nix @@ -7,6 +7,7 @@ , wheel # dependencies +, beartype , einops , torch }: @@ -29,6 +30,7 @@ buildPythonPackage rec { ]; propagatedBuildInputs = [ + beartype einops torch ]; @@ -42,6 +44,7 @@ buildPythonPackage rec { meta = with lib; { description = "Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch"; homepage = "https://github.com/lucidrains/rotary-embedding-torch"; + changelog = "https://github.com/lucidrains/rotary-embedding-torch/releases/tag/${version}"; license = licenses.mit; maintainers = teams.tts.members; };