Commit e4791cb5 authored by happysalada's avatar happysalada Committed by Yt
Browse files

python311Packages.xformers: 0.03 -> 0.0.22.post7 

(cherry picked from commit bdf29479)
parent a6f845c2

pkgs/development/python-modules/xformers/default.nix

+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" ];