mirror of
https://github.com/virchau13/automatic1111-webui-nix.git
synced 2024-08-16 03:00:37 +03:00
42 lines
974 B
Nix
42 lines
974 B
Nix
{ pkgs, isCUDA ? true, ... }:
|
|
|
|
let
|
|
cuda_path = if isCUDA then pkgs.cudatoolkit else "";
|
|
hardware_deps = with pkgs;
|
|
if isCUDA then [
|
|
cudatoolkit
|
|
linuxPackages.nvidia_x11
|
|
xorg.libXi
|
|
xorg.libXmu
|
|
freeglut
|
|
xorg.libXext
|
|
xorg.libX11
|
|
xorg.libXv
|
|
xorg.libXrandr
|
|
zlib
|
|
] else [
|
|
rocm-runtime
|
|
pciutils
|
|
];
|
|
|
|
in
|
|
pkgs.mkShell rec {
|
|
name = "stable-diffusion-webui";
|
|
buildInputs = with pkgs;
|
|
hardware_deps ++ [
|
|
git # The program instantly crashes if git is not present, even if everything is already downloaded
|
|
python310
|
|
stdenv.cc.cc.lib
|
|
stdenv.cc
|
|
ncurses5
|
|
binutils
|
|
gitRepo gnupg autoconf curl
|
|
procps gnumake util-linux m4 gperf unzip
|
|
libGLU libGL
|
|
glib
|
|
];
|
|
LD_LIBRARY_PATH=pkgs.lib.makeLibraryPath buildInputs;
|
|
CUDA_PATH=pkgs.cudatoolkit;
|
|
EXTRA_LDFLAGS="-L${pkgs.linuxPackages.nvidia_x11}/lib";
|
|
}
|