Adding launcher to build. (#2397)

huggingface · Aug 12, 2024 · 730fa00 · 730fa00
1 parent 9c73965
commit 730fa00
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/_launcher.nix b/_launcher.nix
@@ -0,0 +1,18 @@
+{ buildRustPackage, importCargoLock, pkg-config, protobuf, openssl }:
+
+buildRustPackage {
+  name = "text-generation-lancher";
+
+  src = ./.;
+
+  sourceDir = ./launcher;
+
+  cargoLock = {
+    lockFile = ./Cargo.lock;
+  };
+
+  nativeBuildInputs = [ pkg-config ];
+
+  buildInputs = [ openssl.dev protobuf ];
+
+}
diff --git a/flake.nix b/flake.nix
@@ -74,6 +74,9 @@
                 (callPackage ./router.nix {
                   inherit (rustPlatform) buildRustPackage importCargoLock;
                 })
+                (callPackage ./_launcher.nix {
+                  inherit (rustPlatform) buildRustPackage importCargoLock;
+                })
               ]);
 
             venvDir = "./.venv";