feat(nix): Add Docker building configuration

1 files changed, 19 insertions, 0 deletions

diff --git a/gemma-docker.nix b/gemma-docker.nix
new file mode 100644
index 000000000000..69cfdd4c1692
--- /dev/null
+++ b/gemma-docker.nix
@@ -0,0 +1,19 @@
+with import <nixpkgs> {}; 
+let
+  gemma = import ./default.nix;
+  entrypoint = writeScript "entrypoint.sh" ''
+    #!${stdenv.shell}
+    set -e
+    exec ${gemma}/bin/gemma
+  '';
+in dockerTools.buildImage {
+  name = "gemma";
+  contents = gemma; # [ gemma ];
+  config = {
+    Entrypoint = [ entrypoint ];
+    WorkingDir = "/data";
+    Volumes = {
+      "/data" = {};
+    };
+  };
+}