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

View file

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