python310Packages.rotary-embedding-torch: init at 0.3.0

{ 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;

  };

}

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