diff --git a/flake.nix b/flake.nix
index 6275d63..47170b0 100644
--- a/flake.nix
+++ b/flake.nix
@@ -320,7 +320,10 @@
               inputs.nixos-hardware.nixosModules.common-gpu-amd
               inputs.nixos-hardware.nixosModules.common-pc-ssd
               self.nixosModules.server
-              { nixpkgs.config.rocmSupport = true; }
+              {
+                nixpkgs.config.rocmSupport = true;
+                services.ollama.rocmOverrideGfx = "9.0.0";
+              }
 
               ./instances/heimdall/hardware-configuration.nix
               ./instances/heimdall/configuration.nix