From c7aaa3fa1cd2a9221fe2714fa7de0b3a5210a136 Mon Sep 17 00:00:00 2001
From: chn <chn@chn.moe>
Date: Sun, 16 Feb 2025 21:03:04 +0800
Subject: [PATCH] devices.srv2.node0: enable ollama

---
 devices/srv2/node0/default.nix     | 1 +
 modules/system/nixpkgs/default.nix | 3 +++
 2 files changed, 4 insertions(+)

diff --git a/devices/srv2/node0/default.nix b/devices/srv2/node0/default.nix
index 3500dbfa..9f71a6fc 100644
--- a/devices/srv2/node0/default.nix
+++ b/devices/srv2/node0/default.nix
@@ -23,6 +23,7 @@ inputs:
         samba = { enable = true; hostsAllowed = ""; shares = { home.path = "/home"; root.path = "/"; }; };
         groupshare = {};
         hpcstat = {};
+        ollama = {};
       };
     };
     # allow other machine access network by this machine
diff --git a/modules/system/nixpkgs/default.nix b/modules/system/nixpkgs/default.nix
index a1b259ee..75282ffa 100644
--- a/modules/system/nixpkgs/default.nix
+++ b/modules/system/nixpkgs/default.nix
@@ -76,6 +76,9 @@ inputs:
                             { disabledTests = prev.disabledTests or [] ++ [ "test_hyp2f1" ]; });
                         })];
                         rapidjson = prev.rapidjson.overrideAttrs { doCheck = false; };
+                        ctranslate2 = (prev.ctranslate2.override { withCUDA = false; withCuDNN = false; })
+                          .overrideAttrs (prev:
+                            { cmakeFlags = prev.cmakeFlags or [] ++ [ "-DENABLE_CPU_DISPATCH=OFF" ]; });
                       }
                       // inputs.lib.optionalAttrs (nixpkgs.cuda != null)
                       {