added an image generation model
This commit is contained in:
@@ -1,5 +1,10 @@
|
||||
{self, inputs, ...}: {
|
||||
flake.nixosModules.llama = { lib, pkgs, ... }: {
|
||||
{self, inputs, ...}:{
|
||||
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 = [
|
||||
(final: prev: {
|
||||
llama-cpp-cuda = prev.llama-cpp.override {
|
||||
@@ -17,13 +22,18 @@
|
||||
models = {
|
||||
"magidonia" = {
|
||||
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; [
|
||||
llama-cpp-cuda
|
||||
stable-diffusion-cpp-cuda
|
||||
];
|
||||
};
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user