Browse Source

flake : add train-text-from-scratch to flake.nix (#3042)

takov751 2 years ago
parent
commit
ec2a24fedf
1 changed files with 4 additions and 0 deletions
  1. 4 0
      flake.nix

+ 4 - 0
flake.nix

@@ -93,6 +93,10 @@
           type = "app";
           type = "app";
           program = "${self.packages.${system}.default}/bin/quantize";
           program = "${self.packages.${system}.default}/bin/quantize";
         };
         };
+        apps.train-text-from-scratch = {
+          type = "app";
+          program = "${self.packages.${system}.default}/bin/train-text-from-scratch";
+        };
         apps.default = self.apps.${system}.llama;
         apps.default = self.apps.${system}.llama;
         devShells.default = pkgs.mkShell {
         devShells.default = pkgs.mkShell {
           buildInputs = [ llama-python ];
           buildInputs = [ llama-python ];