Unverified Commit e6dad6ee authored by Fabian Affolter's avatar Fabian Affolter Committed by GitHub
Browse files

Merge pull request #280721 from fabaff/rotary-embedding-torch-fix 

python311Packages.rotary-embedding-torch: add beartype
parents 093fc41d 74a7d53e

pkgs/development/python-modules/rotary-embedding-torch/default.nix

+3 −0
Original line number Diff line number Diff line
@@ -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;

  };