python311Packages.rotary-embedding-torch: add beartype

This commit is contained in:
Fabian Affolter
2024-01-13 13:31:52 +01:00
parent 3d673c9b54
commit 74a7d53e68
@@ -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;
};