Skip to content

Commit 32933a4

Browse files
committed
Add in llama-cpp, CUDA edition
1 parent b0dc4ff commit 32933a4

File tree

3 files changed

+21
-48
lines changed

3 files changed

+21
-48
lines changed

flake.lock

Lines changed: 13 additions & 46 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

flake.nix

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,13 +10,13 @@
1010

1111
agenix.url = "github:ryantm/agenix";
1212
agenix.inputs.nixpkgs.follows = "nixpkgs";
13-
agenix.inputs.darwin.follows = "empty";
13+
agenix.inputs.darwin.follows = "nix-darwin";
1414
agenix.inputs.home-manager.follows = "empty";
1515

1616
fh.url = "https://flakehub.com/f/DeterminateSystems/fh/*.tar.gz";
1717
fh.inputs.nixpkgs.follows = "nixpkgs";
1818

19-
llama-cpp.url = "github:ggerganov/llama.cpp";
19+
llama-cpp.url = "github:philiptaron/llama.cpp/nix";
2020
llama-cpp.inputs.nixpkgs.follows = "nixpkgs";
2121

2222
nix-darwin.url = "github:LnL7/nix-darwin";
@@ -32,6 +32,7 @@
3232
agenix = inputs.agenix.overlays.default;
3333
fh = inputs.fh.overlays.default;
3434
nurl = inputs.nurl.overlays.default;
35+
llama-cpp = inputs.llama-cpp.overlays.default;
3536
};
3637
mkConfig = system: {
3738
inherit system;

programs.nix

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -158,6 +158,11 @@
158158

159159
# `nurl` generates Nix fetcher calls from repository URLs
160160
nurl
161+
162+
# `llama-cpp` is a set of programs for running LLMs locally
163+
(llama-cpp-cuda.override {
164+
cudaPackages = pkgs.cudaPackages_12_3;
165+
})
161166
];
162167

163168
users.users.philip.packages = with pkgs; [

0 commit comments

Comments
 (0)