Unverified Commit e7be4bd0 authored by OTABI Tomoya's avatar OTABI Tomoya Committed by GitHub
Browse files

Merge pull request #269065 from happysalada/update_xformers

python311Packages.xformers: 0.03 -> 0.0.22.post7
parents 40316893 bdf29479
Loading
Loading
Loading
Loading
+2 −10
Original line number Diff line number Diff line
@@ -2,12 +2,10 @@
, buildPythonPackage
, pythonOlder
, fetchFromGitHub
, pythonRelaxDepsHook
, which
# runtime dependencies
, numpy
, torch
, pyre-extensions
# check dependencies
, pytestCheckHook
, pytest-cov
@@ -27,7 +25,7 @@
#, flash-attn
}:
let
  version = "0.03";
  version = "0.0.22.post7";
in
buildPythonPackage {
  pname = "xformers";
@@ -40,7 +38,7 @@ buildPythonPackage {
    owner = "facebookresearch";
    repo = "xformers";
    rev = "refs/tags/v${version}";
    hash = "sha256-G8f7tny5B8SAQ6+2uOjhY7nD0uOT4sskIwtTdwivQXo=";
    hash = "sha256-7lZi3+2dVDZJFYCUlxsyDU8t9qdnl+b2ERRXKA6Zp7U=";
    fetchSubmodules = true;
  };

@@ -52,18 +50,12 @@ buildPythonPackage {
  '';

  nativeBuildInputs = [
    pythonRelaxDepsHook
    which
  ];

  pythonRelaxDeps = [
    "pyre-extensions"
  ];

  propagatedBuildInputs = [
    numpy
    torch
    pyre-extensions
  ];

  pythonImportsCheck = [ "xformers" ];