AUTOMATIC1111/stable-diffusion-webui for CUDA and ROCm on NixOS
Go to file
2023-04-15 08:10:16 +08:00
flake.lock feat: initial functionality 2022-12-31 18:04:07 +08:00
flake.nix Add rocm support 2023-04-03 14:07:27 +01:00
impl.nix Better environment switching for CUDA 2023-04-04 11:00:18 +01:00
LICENSE chore: add LICENSE 2022-12-31 18:09:05 +08:00
README.md docs: update README for ROCm support 2023-04-15 08:10:16 +08:00
shell.nix Allow AMD support in shell.nix 2023-04-04 10:57:54 +01:00

AUTOMATIC1111/stable-diffusion-webui for CUDA and ROCm on NixOS

This is literally just a shell.nix/flake.nix for stable-diffusion-webui that also enables CUDA/ROCm on NixOS. This supports both NVIDIA GPUs (using CUDA) and AMD GPUs (using ROCm).

Usage

git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui
git clone https://github.com/virchau13/automatic1111-webui-nix
cp automatic1111-webui-nix/*.nix stable-diffusion-webui/
cd stable-diffusion-webui
git add *.nix
nix develop # or `nix-shell` if you're not using flakes
# just use `./webui.sh` to run it, it'll install all the rest automatically
# follow the tutorials at the original project for setting up Stable Diffusion / GFPGAN / whatever

You might want to switch to high performance mode on battery-powered devices.

Is this completely pure?

This is just a Nix shell for bootstrapping the web UI, not an actual pure flake; the ./webui.sh will still install a bunch of Python packages (into a venv, so not polluting your system) when you run it.

Credits