Add ikllamacpp
This commit is contained in:
parent
75a38bad8a
commit
181d8697f8
1 changed files with 30 additions and 29 deletions
15
flake.nix
15
flake.nix
|
@ -44,8 +44,6 @@
|
|||
(system:
|
||||
let
|
||||
overlays = [
|
||||
llama-cpp.overlays.default
|
||||
ik_llama-cpp.overlays.default
|
||||
rust-overlay.overlays.default
|
||||
devshell.overlays.default
|
||||
(final: prev: {
|
||||
|
@ -102,7 +100,7 @@
|
|||
"gfx1103"
|
||||
];
|
||||
lib = pkgs.lib;
|
||||
myOverride = (oldAttrs: {
|
||||
myOverride = oldAttrs: {
|
||||
# speeds up builts by only building for a needed rocmTargets...
|
||||
cmakeFlags = [
|
||||
(lib.cmakeBool "LLAMA_BUILD_SERVER" true)
|
||||
|
@ -125,20 +123,23 @@
|
|||
(lib.cmakeFeature "CMAKE_HIP_ARCHITECTURES" (builtins.concatStringsSep ";" rocmTargets))
|
||||
(lib.cmakeFeature "AMDGPU_TARGETS" (builtins.concatStringsSep ";" rocmTargets))
|
||||
];
|
||||
});
|
||||
};
|
||||
in
|
||||
{
|
||||
packages =
|
||||
(import ./.nix { inherit pkgs lib config; })
|
||||
// {
|
||||
myikllamacpp = ik_llama-cpp.packages.${system}.default;
|
||||
myllamacpp = pkgs.llamaPackages.llama-cpp.overrideDerivation myOverride;
|
||||
myikllamacpp = ik_llama-cpp.legacyPackages.${system}.llamaPackages.llama-cpp;
|
||||
myikllamacpp-rocm = ik_llama-cpp.legacyPackages.${system}.llamaPackagesRocm.llama-cpp.overrideDerivation myOverride;
|
||||
myllamacpp = llama-cpp.legacyPackages.${system}.llamaPackages.llama-cpp;
|
||||
myllamacpp-rocm = llama-cpp.legacyPackages.${system}.llamaPackagesRocm.llama-cpp.overrideDerivation myOverride;
|
||||
};
|
||||
devShells.default = pkgs.mkShell {
|
||||
packages = with pkgs;
|
||||
[
|
||||
customRustToolchain
|
||||
self.packages.${system}.myllamacpp
|
||||
# self.packages.${system}.myllamacpp
|
||||
self.packages.${system}.myikllamacpp-rocm
|
||||
aider-chat
|
||||
bacon
|
||||
binaryen
|
||||
|
|
Loading…
Add table
Reference in a new issue