···

       
31
       
31
       
 
       
}:

     

       
32
       
32
       
 
       
let

     

       
33
       
33
       
 
       
  inherit (torch) cudaCapabilities cudaPackages cudaSupport;

     

       
34
       
34
       
-
       
  version = "0.0.32.post2";

     

       
34
       
34
       
+
       


     

       
35
       
35
       
+
       
  # version 0.0.32.post2 was confirmed to break CUDA.

     

       
36
       
36
       
+
       
  # Remove this note once the latest published revision "just works".

     

       
37
       
37
       
+
       
  version = "0.0.30";

     

       
35
       
38
       
 
       
in

     

       
36
       
39
       
 
       
buildPythonPackage {

     

       
37
       
40
       
 
       
  pname = "xformers";

     

       
38
       
41
       
 
       
  inherit version;

     

       
39
       
42
       
 
       
  pyproject = true;

     

       
40
       
43
       
 
       


     

       
41
       
41
       
-
       
  disabled = pythonOlder "3.9";

     

       
42
       
42
       
-
       


     

       
43
       
44
       
 
       
  src = fetchFromGitHub {

     

       
44
       
45
       
 
       
    owner = "facebookresearch";

     

       
45
       
46
       
 
       
    repo = "xformers";

     

       
46
       
47
       
 
       
    tag = "v${version}";

     

       
47
       
48
       
 
       
    fetchSubmodules = true;

     

       
48
       
48
       
-
       
    hash = "sha256-B0rvV9EONWQBNhlQ0UTMZ+VtRO/YmMRRlLrVQHGftPE=";

     

       
49
       
49
       
+
       
    hash = "sha256-ozaw9z8qnGpZ28LQNtwmKeVnrn7KDWNeJKtT6g6Q/W0=";

     

       
49
       
50
       
 
       
  };

     

       
50
       
51
       
 
       


     

       
51
       
52
       
 
       
  patches = [ ./0001-fix-allow-building-without-git.patch ];