mirror of
https://github.com/Alexandre1a/NixOSDots.git
synced 2026-03-10 00:09:46 +01:00
25 lines
530 B
Nix
25 lines
530 B
Nix
{ pkgs, ... }:
|
|
|
|
let
|
|
# Détection simple du GPU pour choisir le package Ollama
|
|
gpu =
|
|
if builtins.hasAttr "nvidia" pkgs then
|
|
pkgs.ollama-cuda
|
|
else if builtins.hasAttr "vulkan" pkgs then
|
|
pkgs.ollama-vulkan
|
|
else if builtins.hasAttr "rocm" pkgs then
|
|
pkgs.ollama-rocm
|
|
else
|
|
pkgs.ollama; # CPU fallback
|
|
in
|
|
{
|
|
services = {
|
|
ollama = {
|
|
enable = true;
|
|
loadModels = [ "codegemma:7b-instruct" "codegemma:7b-code" "codegemma:2b" "starcoder2:7b" ];
|
|
package = gpu;
|
|
};
|
|
};
|
|
|
|
}
|