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

invokeai: add wsl variants

Browse source 
This sets the LD_LIBRARY_PATH to /usr/lib/wsl/lib so that invokeai can use a Windows Host's GPU drivers
This commit is contained in:
matthewcroughan 2023-02-13 01:55:35 +00:00
parent d6072e2599
commit a91a1a6db1
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
projects/invokeai/default.nix
View file

@ -10,7 +10,7 @@
builtins.head builtins.head
]; ];
mkInvokeAIVariant = aipython3: aipython3.buildPythonPackage rec { mkInvokeAIVariant = { aipython3, wsl ? false }: aipython3.buildPythonPackage rec {
pname = "InvokeAI"; pname = "InvokeAI";
version = getVersion src; version = getVersion src;
src = inputs.invokeai-src; src = inputs.invokeai-src;
@ -55,6 +55,7 @@
pythonRemoveDeps = [ "clip" "pyreadline3" "flaskwebgui" ]; pythonRemoveDeps = [ "clip" "pyreadline3" "flaskwebgui" ];
pythonRelaxDeps = [ "protobuf" ]; pythonRelaxDeps = [ "protobuf" ];
postFixup = '' postFixup = ''
${lib.optionalString wsl "makeWrapperArgs+=( --set LD_LIBRARY_PATH '/usr/lib/wsl/lib' )"}
chmod +x $out/bin/* chmod +x $out/bin/*
wrapPythonPrograms wrapPythonPrograms
''; '';
@ -68,8 +69,10 @@
in { in {
packages = { packages = {
invokeai-amd = mkInvokeAIVariant aipython3-amd; invokeai-amd = mkInvokeAIVariant { aipython3 = aipython3-amd; };
invokeai-nvidia = mkInvokeAIVariant aipython3-nvidia; invokeai-nvidia = mkInvokeAIVariant { aipython3 = aipython3-nvidia; };
invokeai-amd-wsl = mkInvokeAIVariant { aipython3 = aipython3-amd; wsl = true; };
invokeai-nvidia-wsl = mkInvokeAIVariant { aipython3 = aipython3-nvidia; wsl = true;};
}; };
}; };
} }