diff --git a/nixos/modules/services/home-automation/wyoming/piper.nix b/nixos/modules/services/home-automation/wyoming/piper.nix index e97f99b89ce1..b332c34bd926 100644 --- a/nixos/modules/services/home-automation/wyoming/piper.nix +++ b/nixos/modules/services/home-automation/wyoming/piper.nix @@ -10,6 +10,7 @@ let cfg = config.services.wyoming.piper; inherit (lib) + literalExpression mkOption mkEnableOption mkPackageOption @@ -100,6 +101,15 @@ in default = true; }; + useCUDA = mkOption { + type = bool; + default = config.cudaSupport; + defaultText = literalExpression "config.cudaSupport"; + description = '' + Whether to accelerate the underlying onnxruntime library with CUDA. + ''; + }; + extraArgs = mkOption { type = listOf str; default = [ ]; @@ -165,6 +175,9 @@ in ++ lib.optionals options.streaming [ "--streaming" ] + ++ lib.optionals options.cuda [ + "--use-cuda" + ] ++ options.extraArgs ); CapabilityBoundingSet = "";