1 files changed, 27 insertions, 0 deletions
diff --git a/derivations/flash-attn/flake.nix b/derivations/flash-attn/flake.nix
new file mode 100644
index 0000000..0e09fdf
--- /dev/null
+++ b/derivations/flash-attn/flake.nix
@@ -0,0 +1,27 @@
+# https://github.com/BatteredBunny/nix-ai-stuff/tree/main
+{
+  description = "Gemini CLI flake";
+
+  inputs = {
+    nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable";
+  };
+
+  outputs = {
+    self,
+    nixpkgs,
+  }: let
+    system = "x86_64-linux";
+    pkgs = import nixpkgs {
+      system = system;
+      config = {
+        allowUnfree = true;
+        cudaSupport = true;
+      };
+    };
+  in {
+    packages.${system}.default = pkgs.callPackage ./default.nix {
+      # inherit (pkgs) lib python3Packages fetchFromGitHub symlinkJoin;
+      # pkgs = pkgs;
+    };
+  };
+}