python310Packages.rotary-embedding-torch: init at 0.3.0
Packaged as a dependency for k-diffusion.
This commit is contained in:
parent
3b4c9a0eb7
commit
756eaa6612
|
@ -0,0 +1,48 @@
|
|||
{ lib
|
||||
, buildPythonPackage
|
||||
, fetchFromGitHub
|
||||
|
||||
# build-system
|
||||
, setuptools
|
||||
, wheel
|
||||
|
||||
# dependencies
|
||||
, einops
|
||||
, torch
|
||||
}:
|
||||
|
||||
buildPythonPackage rec {
|
||||
pname = "rotary-embedding-torch";
|
||||
version = "0.3.0";
|
||||
pyproject = true;
|
||||
|
||||
src = fetchFromGitHub {
|
||||
owner = "lucidrains";
|
||||
repo = "rotary-embedding-torch";
|
||||
rev = version;
|
||||
hash = "sha256-fGyBBPfvVq1iZ2m2NNjmHSK+iy76N/09Pt11YDyOyN4=";
|
||||
};
|
||||
|
||||
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;
|
||||
};
|
||||
}
|
|
@ -11509,6 +11509,8 @@ self: super: with self; {
|
|||
|
||||
ropper = callPackage ../development/python-modules/ropper { };
|
||||
|
||||
rotary-embedding-torch = callPackage ../development/python-modules/rotary-embedding-torch { };
|
||||
|
||||
rouge-score = callPackage ../development/python-modules/rouge-score { };
|
||||
|
||||
routeros-api = callPackage ../development/python-modules/routeros-api { };
|
||||
|
|
Loading…
Reference in a new issue