added an image generation model
This commit is contained in:
@@ -7,6 +7,7 @@
|
|||||||
self.nixosModules.llama
|
self.nixosModules.llama
|
||||||
self.nixosModules.localization
|
self.nixosModules.localization
|
||||||
self.nixosModules.applications
|
self.nixosModules.applications
|
||||||
|
self.nixosModules.utilities
|
||||||
];
|
];
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,5 +1,10 @@
|
|||||||
{self, inputs, ...}:{
|
{self, inputs, ...}:{
|
||||||
flake.nixosModules.llama = { lib, pkgs, ... }: {
|
flake.nixosModules.llama = { lib, pkgs, ... }: let
|
||||||
|
llama_cmd = "${pkgs.llama-cpp-cuda}/bin/llama-server --port \${PORT}";
|
||||||
|
sd_cmd = "${pkgs.stable-diffusion-cpp-cuda}/bin/sd-server --listen-port \${PORT}";
|
||||||
|
chat_models = "/var/AI/Models/Chat/ggufs";
|
||||||
|
sd_models = "/var/AI/Models/Art";
|
||||||
|
in {
|
||||||
nixpkgs.overlays = [
|
nixpkgs.overlays = [
|
||||||
(final: prev: {
|
(final: prev: {
|
||||||
llama-cpp-cuda = prev.llama-cpp.override {
|
llama-cpp-cuda = prev.llama-cpp.override {
|
||||||
@@ -17,13 +22,18 @@
|
|||||||
models = {
|
models = {
|
||||||
"magidonia" = {
|
"magidonia" = {
|
||||||
ttl = 3600;
|
ttl = 3600;
|
||||||
cmd = "${pkgs.llama-cpp-cuda}/bin/llama-server --port \${PORT} -m /var/AI/Models/Chat/ggufs/Magidonia-24B-v4.3-Q4_K_M.gguf";
|
cmd = "${llama_cmd} -m ${chat_models}/Magidonia-24B-v4.3-Q4_K_M.gguf";
|
||||||
|
};
|
||||||
|
"cyberrealistic" = {
|
||||||
|
ttl = 3600;
|
||||||
|
cmd = "${sd_cmd} --model ${sd_models}/cyberrealisticPony_semiRealV45.safetensors";
|
||||||
};
|
};
|
||||||
};
|
};
|
||||||
};
|
};
|
||||||
};
|
};
|
||||||
environment.systemPackages = with pkgs; [
|
environment.systemPackages = with pkgs; [
|
||||||
llama-cpp-cuda
|
llama-cpp-cuda
|
||||||
|
stable-diffusion-cpp-cuda
|
||||||
];
|
];
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user