Update modules/nixos/tools/develop/ollama/default.nix
This commit is contained in:
@@ -1,14 +1,13 @@
|
|||||||
{ lib, config, pkgs, ... }:
|
{ lib, config, pkgs, ... }:
|
||||||
let
|
let
|
||||||
ollamaPkg = pkgs.ollama-rocm;
|
ollamaPkg = pkgs.ollama-vulkan;
|
||||||
in{
|
in{
|
||||||
|
|
||||||
config = lib.mkIf (config.syscfg.make.develop) {
|
config = lib.mkIf (config.syscfg.make.develop) {
|
||||||
services.ollama = {
|
services.ollama = {
|
||||||
enable = true;
|
enable = true;
|
||||||
package = ollamaPkg;
|
package = ollamaPkg;
|
||||||
acceleration = "rocm";
|
loadModels = [ ];
|
||||||
loadModels = [ "deepseek-v2:lite" "qwen2.5-coder:7b" "qwen2.5-coder:1.5b" ];
|
|
||||||
syncModels = true;
|
syncModels = true;
|
||||||
};
|
};
|
||||||
environment.systemPackages = with pkgs; [ ollamaPkg ];
|
environment.systemPackages = with pkgs; [ ollamaPkg ];
|
||||||
|
|||||||
Reference in New Issue
Block a user