denis/flake
1
0
Fork 0
mirror of https://github.com/denismhz/flake.git synced 2025-11-09 16:16:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update

Browse source
This commit is contained in:
Denis Manherz 2023-12-02 01:45:55 +01:00
parent 11df303011
commit d12a51a9d5
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
packages/xformers/default.nix
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;

2
projects/automatic1111/package.nix
View file

@ -36,7 +36,7 @@ python3Packages.buildPythonPackage {
taming-transformers-rom1504 taming-transformers-rom1504
timm timm
tomesd tomesd
torch torchWithCuda
transformers transformers
xformers xformers