Adding more kernels to flake. (#2411)

cd9b15d1 · Nicolas Patry · GitHub · 6f4bb4f2 · cd9b15d1 · cd9b15d1
Unverified Commit cd9b15d1 authored Aug 13, 2024 by Nicolas Patry Committed by GitHub Aug 13, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 11 additions and 9 deletions

flake.lock flake.lock +9 -9

flake.nix flake.nix +2 -0

No files found.
--- a/flake.lock
+++ b/flake.lock
@@ -763,11 +763,11 @@
        "treefmt-nix": "treefmt-nix"
      },
      "locked": {
-        "lastModified": 1723343306,
-        "narHash": "sha256-/6sRkPq7/5weX2y0V8sQ29Sz35nt8kyj+BsFtkhgbJE=",
+        "lastModified": 1723512448,
+        "narHash": "sha256-VSTtxGKre1p6zd6ACuBmfDcR+BT9+ml8Y3KrSbfGFYU=",
        "owner": "nix-community",
        "repo": "poetry2nix",
-        "rev": "4a1c112ff0c67f496573dc345bd0b2247818fc29",
+        "rev": "ed52f844c4dd04dde45550c3189529854384124e",
        "type": "github"
      },
      "original": {
@@ -897,11 +897,11 @@
        ]
      },
      "locked": {
-        "lastModified": 1723429325,
-        "narHash": "sha256-4x/32xTCd+xCwFoI/kKSiCr5LQA2ZlyTRYXKEni5HR8=",
+        "lastModified": 1723515680,
+        "narHash": "sha256-nHdKymsHCVIh0Wdm4MvSgxcTTg34FJIYHRQkQYaSuvk=",
        "owner": "oxalica",
        "repo": "rust-overlay",
-        "rev": "65e3dc0fe079fe8df087cd38f1fe6836a0373aad",
+        "rev": "4ee3d9e9569f70d7bb40f28804d6fe950c81eab3",
        "type": "github"
      },
      "original": {
@@ -1035,11 +1035,11 @@
        "nixpkgs": "nixpkgs_7"
      },
      "locked": {
-        "lastModified": 1723450799,
-        "narHash": "sha256-cuT/ce7R2D5Lx6Ted4YS4y+WrAAOXQFAbzLsM1vtPo8=",
+        "lastModified": 1723532088,
+        "narHash": "sha256-6h/P/BkFDw8txlikonKXp5IbluHSPhHJTQRftJLkbLQ=",
        "owner": "danieldk",
        "repo": "tgi-nix",
-        "rev": "29f9b45bd613eced65c4a5241a00aa9346f63d90",
+        "rev": "32335a37ce0f703bab901baf7b74eb11e9972d5f",
        "type": "github"
      },
      "original": {

--- a/flake.nix
+++ b/flake.nix
@@ -66,6 +66,7 @@
                venvShellHook
                pip

+                causal-conv1d
                click
                einops
                fbgemm-gpu
@@ -79,6 +80,7 @@
                grpcio-tools
                hf-transfer
                loguru
+                mamba-ssm
                marlin-kernels
                opentelemetry-api
                opentelemetry-exporter-otlp