Package request: python3Packages.local-attention

[hakan-demirli]

Project description

Local attention, window with lookback, for language modeling

Metadata

• homepage URL: https://github.com/lucidrains/local-attention
• source URL: https://github.com/lucidrains/local-attention
• license: mit
• platforms: unix, linux, darwin

---

Note for maintainers: Please tag this issue in your PR.

---

Add a :+1: reaction to issues you find important.

[hakan-demirli]

Works on nixos-unstable

{
  lib,
  python3Packages,
  fetchurl,
}:
python3Packages.buildPythonPackage rec {
  pname = "local-attention";
  version = "1.9.15";

  src = fetchurl {
    url = "https://files.pythonhosted.org/packages/86/88/1a952205a1f40c38ff3c956d245ff4f0579208ba5859c05124b2d937dcc5/local_attention-1.9.15-py3-none-any.whl";
    sha256 = "sha256-0wVb24fBqKaMZ5W4SbRT6pwq2w3EJtcvcb1T+Evm4Ns=";
  };

  format = "wheel";

  propagatedBuildInputs = with python3Packages; [
    # numpy
    # scipy
    # psutil
    # # clipboard
    # matplotlib
    # keyboard
  ];

  meta = with lib; {
    description = "Local attention, window with lookback, for language modeling";
    homepage = "https://github.com/lucidrains/local-attention";
    license = licenses.mit;
  };
}