From 7fe6438fc6c2a7d5dfb98cccfca388fab85cb351 Mon Sep 17 00:00:00 2001
From: Nikita Wootten <nikita.wootten@gmail.com>
Date: Sat, 2 Dec 2023 10:28:10 -0500
Subject: [PATCH] Add option to enable beta driver

---
 hostModules/personal/nvidia.nix | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/hostModules/personal/nvidia.nix b/hostModules/personal/nvidia.nix
index 7109a6f..c70d2a7 100644
--- a/hostModules/personal/nvidia.nix
+++ b/hostModules/personal/nvidia.nix
@@ -10,6 +10,7 @@ in
       description = "If true, enable NVidia settings";
       default = true;
     };
+    betaDriver = lib.mkEnableOption "enable beta driver";
   };
 
   config = lib.mkIf cfg.enable {
@@ -27,9 +28,11 @@ in
       modesetting.enable = true;
       open = false;
       nvidiaSettings = !cfg.headless;
-      package = config.boot.kernelPackages.nvidiaPackages.stable;
+      package = if cfg.betaDriver
+        then config.boot.kernelPackages.nvidiaPackages.beta
+        else config.boot.kernelPackages.nvidiaPackages.stable;
     };
 
     virtualisation.docker.enableNvidia = true;
   };
-}
\ No newline at end of file
+}