cafkafk/nixpkgs
cafkafk/nixpkgs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2

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

Browse source 
python311Packages.rotary-embedding-torch: add beartype
This commit is contained in:
Fabian Affolter 2024-01-14 23:54:55 +01:00 committed by GitHub
commit e6dad6ee45
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -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;
};