Compare commits

...

3 commits

Author SHA1 Message Date
Denis Manherz d12a51a9d5 Update 2023-12-02 01:45:55 +01:00
Denis Manherz 11df303011 Update 2023-12-02 01:40:14 +01:00
Denis Manherz fb68dcf306 Update 2023-12-02 01:26:06 +01:00
2 changed files with 4 additions and 3 deletions

View file

@ -2,7 +2,7 @@
# If you run pynixify again, the file will be either overwritten or
# deleted, and you will lose the changes you made to it.
{ buildPythonPackage, fetchPypi, lib, torch, numpy, pyre-extensions, pythonRelaxDepsHook, which }:
{ buildPythonPackage, fetchPypi, lib, torchWithCuda, numpy, pyre-extensions, pythonRelaxDepsHook, which }:
buildPythonPackage rec {
pname = "xformers";
@ -14,7 +14,7 @@ buildPythonPackage rec {
};
nativeBuildInputs = [ pythonRelaxDepsHook which ];
pythonRelaxDeps = [ "pyre-extensions" ];
propagatedBuildInputs = [ torch numpy pyre-extensions /*triton*/ ];
propagatedBuildInputs = [ torchWithCuda numpy pyre-extensions /*triton*/ ];
# TODO FIXME
doCheck = false;

View file

@ -36,7 +36,7 @@ python3Packages.buildPythonPackage {
taming-transformers-rom1504
timm
tomesd
torch
torchWithCuda
transformers
xformers
@ -140,4 +140,5 @@ python3Packages.buildPythonPackage {
#civit-ai browser + couple of deps
#animatediff --> needs deforum for frame interpolation
#deforum
#controlnet
}