mirror of
https://github.com/denismhz/flake.git
synced 2025-11-09 16:16:23 +01:00
Compare commits
3 commits
f878b1291a
...
d12a51a9d5
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
d12a51a9d5 | ||
|
|
11df303011 | ||
|
|
fb68dcf306 |
|
|
@ -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;
|
||||
|
|
|
|||
|
|
@ -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
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in a new issue