Commit: fe9d9764ede355f4b7bde24e6084a62e1845781d - zirpu (git)

Emergent generative agents

Commit MetaInfo

• Révision: fe9d9764ede355f4b7bde24e6084a62e1845781d (tree)
• l'heure: 2023-04-18 07:12:37
• Auteur: Corbin <cds@corb...>
• Commiter: Corbin

Message de Log

Initial swing at packaging rwkv.cpp.

Change Summary

• modified: flake.nix

Modification

--- a/flake.nix
+++ b/flake.nix
@@ -14,6 +14,27 @@
         py = pkgs.python310.withPackages (ps: with ps; [
           faiss irc transformers torch
         ]);
+        rwkv = pkgs.stdenv.mkDerivation {
+          name = "rwkv.cpp";
+          version = "0.0.1";
+
+          src = pkgs.fetchFromGitHub {
+            owner = "saharNooby";
+            repo = "rwkv.cpp";
+            rev = "84e0698f2b16955ac82f6dfcafa76b13f27d97c5";
+            sha256 = "sha256-IJ6vq0g8lKlefAIsIa4LFMM+9mM+NXx6escBraC5JDA=";
+            fetchSubmodules = true;
+          };
+
+          buildInputs = with pkgs; [ cmake ];
+
+          cmakeFlags = [ "-DBUILD_SHARED_LIBS=ON" ];
+
+          installPhase = ''
+            mkdir -p $out/lib/
+            cp librwkv.so $out/lib/
+          '';
+        };
       in {
         packages.default = pkgs.stdenv.mkDerivation {
           name = "zirpu";
@@ -21,6 +42,8 @@
 
           src = ./src;
 
+          buildInputs = [ rwkv ];
+
           patchPhase = ''
             sed -i -e 's,@LLAMA@,${llama}/bin/,g' gens/camelid.py
             sed -i -e 's,/usr/bin/env ,${py}/bin/,' *.py