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:
|
(system:
|
||||||
let
|
let
|
||||||
overlays = [
|
overlays = [
|
||||||
llama-cpp.overlays.default
|
|
||||||
ik_llama-cpp.overlays.default
|
|
||||||
rust-overlay.overlays.default
|
rust-overlay.overlays.default
|
||||||
devshell.overlays.default
|
devshell.overlays.default
|
||||||
(final: prev: {
|
(final: prev: {
|
||||||
|
@ -102,7 +100,7 @@
|
||||||
"gfx1103"
|
"gfx1103"
|
||||||
];
|
];
|
||||||
lib = pkgs.lib;
|
lib = pkgs.lib;
|
||||||
myOverride = (oldAttrs: {
|
myOverride = oldAttrs: {
|
||||||
# speeds up builts by only building for a needed rocmTargets...
|
# speeds up builts by only building for a needed rocmTargets...
|
||||||
cmakeFlags = [
|
cmakeFlags = [
|
||||||
(lib.cmakeBool "LLAMA_BUILD_SERVER" true)
|
(lib.cmakeBool "LLAMA_BUILD_SERVER" true)
|
||||||
|
@ -125,20 +123,23 @@
|
||||||
(lib.cmakeFeature "CMAKE_HIP_ARCHITECTURES" (builtins.concatStringsSep ";" rocmTargets))
|
(lib.cmakeFeature "CMAKE_HIP_ARCHITECTURES" (builtins.concatStringsSep ";" rocmTargets))
|
||||||
(lib.cmakeFeature "AMDGPU_TARGETS" (builtins.concatStringsSep ";" rocmTargets))
|
(lib.cmakeFeature "AMDGPU_TARGETS" (builtins.concatStringsSep ";" rocmTargets))
|
||||||
];
|
];
|
||||||
});
|
};
|
||||||
in
|
in
|
||||||
{
|
{
|
||||||
packages =
|
packages =
|
||||||
(import ./.nix { inherit pkgs lib config; })
|
(import ./.nix { inherit pkgs lib config; })
|
||||||
// {
|
// {
|
||||||
myikllamacpp = ik_llama-cpp.packages.${system}.default;
|
myikllamacpp = ik_llama-cpp.legacyPackages.${system}.llamaPackages.llama-cpp;
|
||||||
myllamacpp = pkgs.llamaPackages.llama-cpp.overrideDerivation myOverride;
|
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 {
|
devShells.default = pkgs.mkShell {
|
||||||
packages = with pkgs;
|
packages = with pkgs;
|
||||||
[
|
[
|
||||||
customRustToolchain
|
customRustToolchain
|
||||||
self.packages.${system}.myllamacpp
|
# self.packages.${system}.myllamacpp
|
||||||
|
self.packages.${system}.myikllamacpp-rocm
|
||||||
aider-chat
|
aider-chat
|
||||||
bacon
|
bacon
|
||||||
binaryen
|
binaryen
|
||||||
|
|
Loading…
Add table
Reference in a new issue