From 11ab758490eb269938e105e06db37b5ef89355f2 Mon Sep 17 00:00:00 2001
From: Max <max@privatevoid.net>
Date: Fri, 3 Mar 2023 15:33:28 +0100
Subject: [PATCH] invokeai: add NixOS option for model precision

---
 projects/invokeai/nixos/default.nix | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/projects/invokeai/nixos/default.nix b/projects/invokeai/nixos/default.nix
index f5176a0..70833ad 100644
--- a/projects/invokeai/nixos/default.nix
+++ b/projects/invokeai/nixos/default.nix
@@ -56,6 +56,12 @@ in
 
     nsfwChecker = mkEnableOption "the NSFW Checker";
 
+    precision = mkOption {
+      description = "Set model precision.";
+      default = "auto";
+      type = types.enum [ "auto" "float32" "autocast" "float16" ];
+    };
+
     extraArgs = mkOption {
       description = "Extra command line arguments.";
       default = [];
@@ -74,6 +80,7 @@ in
       "--root_dir" cfg.dataDir
       "--max_loaded_models" cfg.maxLoadedModels
       (yesno cfg.nsfwChecker "nsfw_checker")
+      "--precision" cfg.precision
     ] ++ cfg.extraArgs;
     initialModelsPath = "${cfg.package}/${cfg.package.pythonModule.sitePackages}/invokeai/configs/INITIAL_MODELS.yaml";
   in mkIf cfg.enable {