invokeai: add a NixOS module for AMD-specifc quirks

This commit is contained in:
Max 2023-03-02 23:18:20 +01:00
parent e110b678dc
commit 46e6505660
2 changed files with 13 additions and 0 deletions

View file

@ -29,6 +29,7 @@
invokeai-amd = { invokeai-amd = {
imports = [ imports = [
config.flake.nixosModules.invokeai config.flake.nixosModules.invokeai
./nixos/amd.nix
(packageModule "invokeai-amd") (packageModule "invokeai-amd")
]; ];
}; };

View file

@ -0,0 +1,12 @@
{ pkgs, ... }:
{
systemd = {
# Allow "unsupported" AMD GPUs
services.invokeai.environment.HSA_OVERRIDE_GFX_VERSION = "10.3.0";
# HACK: The PyTorch build we use on ROCm wants this to exist
tmpfiles.rules = [
"L+ /opt/amdgpu - - - - ${pkgs.libdrm}"
];
};
}