python311Packages.xformers: 0.03 -> 0.0.22.post7

, buildPythonPackage

, pythonOlder

, fetchFromGitHub

, pythonRelaxDepsHook

, which

# runtime dependencies

, numpy

, torch

, pyre-extensions

# check dependencies

, pytestCheckHook

, pytest-cov

#, flash-attn

}:

let

  version = "0.03";

  version = "0.0.22.post7";

in

buildPythonPackage {

  pname = "xformers";

    owner = "facebookresearch";

    repo = "xformers";

    rev = "refs/tags/v${version}";

    hash = "sha256-G8f7tny5B8SAQ6+2uOjhY7nD0uOT4sskIwtTdwivQXo=";

    hash = "sha256-7lZi3+2dVDZJFYCUlxsyDU8t9qdnl+b2ERRXKA6Zp7U=";

    fetchSubmodules = true;

  };

  '';

  nativeBuildInputs = [

    pythonRelaxDepsHook

    which

  ];

  pythonRelaxDeps = [

    "pyre-extensions"

  ];

  propagatedBuildInputs = [

    numpy

    torch

    pyre-extensions

  ];

  pythonImportsCheck = [ "xformers" ];