Something went wrong. Try again.
lol
Something went wrong. Try again.
1.0 kB · 48 lines
Nix
at master
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849{ lib, buildPythonPackage, einops, fetchFromGitHub, hyper-connections, pytestCheckHook, setuptools, torch,}:
buildPythonPackage rec { pname = "local-attention"; version = "1.11.2"; pyproject = true;
src = fetchFromGitHub { owner = "lucidrains"; repo = "local-attention"; tag = version; hash = "sha256-2gBPALJAflLf7Y8L5wnNw4fHcvIOKjOncLsebkhrYkU="; };
postPatch = '' substituteInPlace setup.py \ --replace-fail "'pytest-runner'," "" '';
build-system = [ setuptools ];
dependencies = [ einops hyper-connections torch ];
nativeCheckInputs = [ pytestCheckHook ];
pythonImportsCheck = [ "local_attention" ];
meta = { description = "Module for local windowed attention for language modeling"; homepage = "https://github.com/lucidrains/local-attention"; changelog = "https://github.com/lucidrains/local-attention/releases/tag/${src.tag}"; license = lib.licenses.mit; maintainers = with lib.maintainers; [ fab ]; };}